P6475 [NOI Online #2 入门组] 建设城市 题解

一道排列组合题，还是我的排列组合太菜了……

选必三重修警告

先说明一个东西：对于 $n$ 个楼房，高度限制为 $m$，那么高度不下降的方案数为 $C_{n+m-1}^{m-1}$。

理由：$n$ 个楼房选高度，变成 $n+m$ 个楼房选 $m$ 个高度，并且每个高度必须有得选，隔板法即可，$n+m-1$ 个空隙插入 $m-1$ 个板。

题目中一个不上升和一个不下降可以转化为两个不上升，接下来分类讨论 $x,y$ 是否在同侧也就是同个序列里面。

---

先看 $x,y$ 同侧（钦定 $x<y$），此时有一个序列方案数就是 $C_{n+m-1}^{m-1}$，另一个方案数可以将 $[x,y]$ 缩成一个点，这块方案数是 $C_{n-(y-x)+m-1}^{m-1}$，总方案数乘起来即可。

---

然后是 $x,y$ 异侧，这一块需要枚举 $x,y$ 的高度，设为 $k$，然后对两个序列分别讨论，注意这里将 $x,y$ 全部缩到 $[1,n]$ 里面，要讨论的是两个数量为 $n$ 的楼房的方案数。

这里以 $x$ 为例，左边方案数 $C_{x-1+k-1}^{k-1}$，右边方案数 $C_{(n-x)+(m-k+1)-1}^{(m-k+1)-1}$，然后乘起来。对于 $y$ 也如法炮制，最后四项统一乘起来就是高度为 $k$ 的方案数，最后求和即可。

注意要 $O(n)$ 求阶乘逆元不然你会被卡，以及处理阶乘逆元的上限是 $n+m-1$，不要处理小了。

代码里记得转换 $x,y$ 以及上面的组合数括号不要拆掉，拆掉了反而容易错你还不知道错在哪（

GitHub：CodeBase-of-Plozia

Code：

/*
========= Plozia =========
	Author:Plozia
	Problem:P6475 [NOI Online #2 入门组] 建设城市
	Date:2022/3/17
========= Plozia =========
*/

#include <bits/stdc++.h>

typedef long long LL;
const int MAXN = 2e5 + 5;
const LL P = 998244353;
int n, m, x, y;
LL fact[MAXN], inv[MAXN], ans;

int Read()
{
	int sum = 0, fh = 1; char ch = getchar();
	for (; ch < '0' || ch > '9'; ch = getchar()) fh -= (ch == '-') << 1;
	for (; ch >= '0' && ch <= '9'; ch = getchar()) sum = sum * 10 + (ch ^ 48);
	return sum * fh;
}
int Max(int fir, int sec) { return (fir > sec) ? fir : sec; }
int Min(int fir, int sec) { return (fir < sec) ? fir : sec; }
LL ksm(LL a, LL b = P - 2, LL p = P)
{
	LL s = 1 % p;
	for (; b; b >>= 1, a = a * a % p)
		if (b & 1) s = s * a % p;
	return s;
}

LL C(int n, int m)
{
	if (n < m) return 0;
	return fact[n] * inv[m] % P * inv[n - m] % P;
}

void Work1()
{
	if (x > n) x = (2 * n + 1 - x), y = (2 * n + 1 - y);
	if (x > y) std::swap(x, y);
	for (int i = 1; i <= m; ++i)
		ans = (ans + C(n + m - 1, m - 1) * C((x - 1) + i - 1, i - 1) % P * C((n - (y + 1) + 1) + (m - i + 1) - 1, (m - i + 1) - 1) % P) % P;
}

void Work2()
{
	if (x > n) x = (n - (x - n) + 1);
	if (y > n) y = (n - (y - n) + 1);
	for (int i = 1; i <= m; ++i)
		ans = (ans + C((x - 1) + i - 1, i - 1) * C((n - (x + 1) + 1) + (m - i + 1) - 1, (m - i + 1) - 1) % P * C((y - 1) + i - 1, i - 1) % P * C((n - (y + 1) + 1) + (m - i + 1) - 1, (m - i + 1) - 1) % P) % P;
}

int main()
{
	m = Read(), n = Read(), x = Read(), y = Read(); fact[0] = inv[0] = 1;
	for (int i = 1, tmp = Max(n, m); i <= 2 * tmp + 2; ++i) fact[i] = fact[i - 1] * i % P;
	inv[Max(n, m) * 2 + 2] = ksm(fact[Max(n, m) * 2 + 2]);
	for (int i = Max(n, m) * 2 + 1; i >= 1; --i) inv[i] = inv[i + 1] * (i + 1) % P;
	if ((x <= n && y <= n) || (x > n && y > n)) Work1();
	else Work2();
	printf("%lld\n", ans); return 0;
}