洛谷省选计划后期 2021 数数题 题解

题目传送门 然而大部分人基本不能看……

题目大意： 好像也不方便写……

题解

大概就是给以后的自己看的东西。

画一个矩阵，第 $i$ 行第 $j$ 列为 $1$ 表示 $i$ 位置可以选 $j$ 这个数，那么画出来就是个正方形左下角缺了个 $S$ 大小的直角三角形，右上角缺了个 $T$ 大小的直角三角形。问题变成在 $1$ 位置上放 $n$ 个棋子，两两棋子行列互不相同。

假设 $S$ 为 $0$，那么剩下问题的答案就是 $(n-T{)}^T(n-T)!$，你从 $1$ 到 $n$ 考虑每个行有多少种放法即可。

假如 $S>0$，考虑容斥，假设那个左下角的缺口里放了至少 $i$ 个棋子，那么剩下的棋子相当于 $S=0,n^{\prime }=n-i$ 的问题（也就是上面那个），令 $c(S,i)$ 表示大小为 $S$ 的直角三角形里放 $i$ 个棋子的方案数，答案为：
$$\sum _{i=0}^S(-1{)}^{i}c(S,i)(n-i-T{)}^T(n-i-T)!$$

然后就是 $c$ 的求法，对于 $c(n,m)$，看最左边，也就是高为 $n$ 的那一列，考虑这一列上有没有棋子，那么就可以得到递推式：
$$c(n,m)=c(n-1,m)+(n-m+1)c(n-1,m-1)$$

这个东西很像第二类斯特林数的递推式：
$$S(n,m)=S(n-1,m-1)+mS(n-1,m)$$

但似乎又难以联系起来？如果将这个转移方程画在表格上，就会出现神奇的事情（左边是 $S$，右边是 $c$）：

这样看也许还看不出什么，如果你将 $c$ 的表格从左对齐变成右对齐，然后 $S$ 的转移中，将 $m$ 看做这个格子到左边界的距离，$c$ 的转移中 $n-m+1$ 看做这个格子到右边界的距离，你会发现，这两个表格的转移是完全对称的！

这意味着 $c(n,m)=S(n,n-m)$ 吗？事实上并不，因为你还要考虑 $c,S$ 的边界问题，$S$ 的 $n,m$ 从 $1$ 开始，而 $c$ 的 $n,m$ 从 $0$ 开始，所以实际上是 $c(n,m)=S(n+1,n-m+1)$。

这样你再套一个求一行第二类斯特林数板子就做完了，时间复杂度 $O(n\log n)$。

代码如下：

#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
#define maxn 530000
#define mod 998244353
#define bin(x) (1<<(x))

int n,S,T,F[maxn],G[maxn];
int ksm(int x,int y){int re=1;for(;(y&1?re=1ll*re*x%mod:0),y;y>>=1,x=1ll*x*x%mod);return re;}
int w[maxn],inv[maxn];void prep(int lg){int N=bin(lg);
	inv[1]=1;for(int i=2;i<=N;i++)inv[i]=1ll*(mod-mod/i)*inv[mod%i]%mod;
	for(int i=1,wn;i<N;i<<=1){
		w[i]=1;wn=ksm(3,(mod-1)/(i<<1));
		for(int j=1;j<i;j++)w[i+j]=1ll*w[i+j-1]*wn%mod;
	}
}
int limit,rev[maxn];
void InitRev(int lg){for(int i=1;i<bin(lg);i++)rev[i]=(rev[i>>1]>>1)|((i&1)<<(lg-1));}
void reduce(int &x){x+=x>>31&mod;}
void ntt(int *f,int lg,int type=0){
	limit=bin(lg);if(type)reverse(f+1,f+limit);
	for(int i=1;i<limit;i++)if(i<rev[i])swap(f[i],f[rev[i]]);
	for(int mid=1,t;mid<limit;mid<<=1)for(int j=0;j<limit;j+=(mid<<1))for(int i=0;i<mid;i++)
	{t=1ll*f[j+i+mid]*w[mid+i]%mod;reduce(f[j+i+mid]=f[j+i]-t);reduce(f[j+i]+=t-mod);}
	if(type)for(int i=0;i<limit;i++)f[i]=1ll*f[i]*inv[limit]%mod;
}
void NTT(int *f,int *g,int ln,int Mul=2){
	int lg=ceil(log2(ln*Mul));InitRev(lg);ntt(f,lg);ntt(g,lg);
	for(int i=0;i<bin(lg);i++)f[i]=1ll*f[i]*g[i]%mod;ntt(f,lg,1);
}
void get_Stirling(int n){
	prep(ceil(log2(n+1<<1)));
	for(int i=0,inv_fac=1;i<=n;i++){
		F[i]=1ll*ksm(i,n)*inv_fac%mod;
		G[i]=i&1?mod-inv_fac:inv_fac;
		inv_fac=1ll*inv_fac*inv[i+1]%mod;
	}
	NTT(F,G,n+1);
}

int main()
{
	scanf("%d %d %d",&n,&S,&T);
	get_Stirling(S+1);
	static int fac[maxn],ans=0;
	fac[0]=1;for(int i=1;i<=n;i++)fac[i]=1ll*fac[i-1]*i%mod;
	for(int i=0,val;i<=S;i++){
		val=1ll*F[S-i+1]*ksm(n-i-T,T)%mod*fac[n-i-T]%mod;
		reduce(ans-=i&1?val:mod-val);
	}
	printf("%d",ans);
}