NOI Online #2 普及组 第三题：建设城市

前言

这道题是我今天刷的，意为完成Noi Online #2的普及组所有题的复盘

这道题是一道$\text{提高+/省选-}$的题目，也不是算太难

题目

传送门

解析

前置知识：快速幂，组合数学，逆元

---

我们首先设一个函数$f(i,j)$表示值域为$1+x\sim j+x|x\in N$（可以证明不管$x$为多少都对答案没有多大关系），序列长度为$i$的序列的排列方案。

然后由插板法可得

$$f(i,j)=\left(\genfrac{}{}{0ex}{}{i+j-1}{j-1}\right)$$

对于答案的统计，需要分情况讨论

• $x\le n,y>n$

对于这种情况我们可以做一个图示：

在此时，这个序列便分为四个区间：$1\sim x-1,x+1\sim n,n+1\sim y-1,y+1\sim n\ast 2$

通过枚举$m$以及乘法原理，我们可以得到这种情况的答案：

通过枚举mmm以及乘法原理，我们可以得到这种情况的答案：

图示：

$ans=\sum _{i=1}^{m}f(x-1,i)\ast f(n-x,m-i+1)\ast f(y-n-1,m-i+1)\ast f(2\ast n-y,i)$

• $n<x<y$或$x<y<=n$

图示：

注意到$x$与$y$是相等的，所以我们可以把区间$x\sim y$合并成一个点。

又根据乘法原理，我们可以$O(1)$计算出答案：

$$ans=f(n,m)\times f(n+x-y,m)$$

但问题来了：

$\text{如何O(1)}计算f函数？？？？$

我们之前提到过$f(i,j)$就是$\left(\genfrac{}{}{0ex}{}{i+j-1}{j-1}\right)$所以求$f$函数又转化为求组合数

而众所周知，

$$\left(\genfrac{}{}{0ex}{}{n}{m}\right)=\frac{n!}{m!(n-m)!}$$

所以，

$$f(i,j)=\frac{(i+j-1)!}{i!(j-1)!}$$

我们现在剩下的任务就是求$1!\sim 200000!$的逆元（$n+m$最大为$200000$，当然你也可以只预处理到$n+m$）。

接下来介绍两种求逆元的方法：

• 无需理解，会逆元就行

学过都知道，一个数$x$的逆元就是$x^{mod-2}$（不会的记得这个公式就行了）

所以在循环的时候，直接每次利用求出的frafrafra数组（也就是阶乘）以及快速幂进行求逆元就行了

时间复杂度$O(n\log n)$

代码片-1

void prework()
{
	fac[0]=inv[0]=1;
	for(int i=1;i<=200000;i++)fac[i]=(fac[i-1]*i)%mod;
	for(int i=1;i<=200000;i++)inv[i]=quick(fac[i],mod-2);
}

首先我们还需要求出frafrafra数组。

然后求出$inv[n+m]$。

最后后从后往前循环，每次求$inv[i]$（阶乘逆元）时，将$inv[i+1]$乘以$(i+1)$即可得到$inv[i]$。

证明？

$$(n+1)!\times \frac{1}{n+1}=n!$$

时间复杂度为$O(n)$

代码片-2

void prework()
{
	fac[0]=inv[0]=1;
	for(int i=1;i<=200000;i++)fac[i]=(fac[i-1]*i)%mod;
	inv[200000]=quick(fac[200000],mod-2);
	for(int i=199999;i;i--)inv[i]=inv[i+1]*(i+1)%mod;
}

代码片-3

最后是$f$函数的代码

LL Func(LL x,LL y)
{
	return fac[y+x-1]*inv[x]%mod*inv[y-1]%mod;
}

代码

#include<bits/stdc++.h>
using namespace std;
#define LL long long
const int mod=998244353;
const int Maxn=200001;
LL fac[Maxn],inv[Maxn],n,m,ans;
LL qp(LL a,LL b)
{
	LL xx=a,yy=1;
	while(b)
	{
		if(b&1) yy=yy*xx%mod;
		xx=xx*xx%mod;
		b/=2;
	}
	return yy;
}
LL Func(LL x,LL y)
{
	return fac[y+x-1]*inv[x]%mod*inv[y-1]%mod;
}
int main()
{
	LL x,y;
	fac[0]=inv[0]=1;
	for(int i=1;i<=200000;i++)fac[i]=(fac[i-1]*i)%mod;
	inv[200000]=qp(fac[200000],mod-2);
	for(int i=199999;i;i--)inv[i]=inv[i+1]*(i+1)%mod;
	scanf("%lld%lld%lld%lld",&m,&n,&x,&y);
	if(x<=n&&y>n)
	{
		for(int i=1;i<=m;i++)
		{
			ans=(ans+Func(x-1,i)*Func(n-x,m-i+1)%mod*Func(2*n-y,i)%mod*Func(y-n-1,m-i+1))%mod;
		}
	}else{
		ans=Func(n,m)*Func(n+x-y,m)%mod;
	}
	printf("%lld",ans);
	return 0;
}