【UOJ #390】【UNR #3】百鸽笼（指数型生成函数，二项式定理）

题面

🔗

限制

题解

参考

我们将题目等价为：每次随机任意一个位置放人，当人数$\ge a_i$ 时认为 $i$ 放满，一直放，询问 $i$ 成为最后放满人的一列的概率。

我们限定 $I$ 为最后一列，那么其他列就必须放满，合法的放人序列（到 $I$ 填满为止）可以用指数型生成函数（EGF）计算。（放心，不会很难理解的）

我们为了方便表示，令
$$G_k(x)=\sum _{i=0}^{a_k-1}\frac{x^i}{i!}$$

其他列的 EGF 就是 $e^x-G_k(x)$ ，第 $I$ 列则是 $\frac{x^{a_k-1}}{(a_k-1)!}$ ，那么 $I$ 的除去最后一项的合法序列 EGF 就是
$$F(x)=\frac{x^{a_k-1}}{(a_k-1)!}\prod _{i\ne I}(e^x-G_i(x))$$

第 $I$ 列的答案就是所有合法序列的概率之和（由于除去了最后一项，$\frac{1}{n}$ 的幂为长度+1）：
$$\sum _{i=0}^{\infty }\frac{i!\cdot [x^i]F(x)}{n^{i+1}}$$

由于 $e^x$ 长度是无穷的，我们就把 $F(x)$ 当成 $x,e^x$ 的二元生成函数，处理出每个 $x^i(e^x{)}^j$ 的系数 $f_{i,j}$ ，这个可以背包DP做。然后我们单独算每个 $f_{i,j}{x}^i(e^x{)}^j$ 的贡献：
$$\begin{gathered} \frac{f_{i,j}}{n^{i+1}}\sum _{d=0}^{\infty }(d+i)!\frac{j^d}{d!n^d} \\ =\frac{f_{i,j}\cdot i!}{n^{i+1}}\sum _{d=0}^{\infty }(\frac{j}{n}{)}^d(\genfrac{}{}{0ex}{}{d+i}{i}) \end{gathered}$$

右边那个形式感觉很有规律，其实它是广义二项式展开：
$$(x+y{)}^{-n}=\sum _{i=0}^{\infty }(-1{)}^i\left(\genfrac{}{}{0ex}{}{n-1+i}{n-1}\right)x^i{y}^{-n-i}$$

所以上式可以转化为
$$\begin{gathered} \frac{f_{i,j}\cdot i!}{n^{i+1}}\sum _{d=0}^{\infty }(-\frac{j}{n}{)}^d(-1{)}^d\left(\genfrac{}{}{0ex}{}{d+i}{i}\right) \\ =\frac{f_{i,j}\cdot i!}{n^{i+1}}(1-\frac{j}{n}{)}^{-(i+1)} \\ =\frac{f_{i,j}\cdot i!}{n^{i+1}}(\frac{n}{n-j}{)}^{i+1} \\ =\frac{f_{i,j}\cdot i!}{(n-j{)}^{i+1}} \end{gathered}$$

如此简洁。

但是，我们对于每个 $I$ 都求一次背包DP刚好要超时。由于DP过程只有加减法，我们可以先处理出所有列的背包DP，然后再去掉 $I$ （假想把 $I$ 放到最后，按照DP的转移反着做一次）。这样复杂度就对了，$O(n^3\sum a_i)$ 。

CODE

#include<map>
#include<set>
#include<cmath>
#include<ctime>
#include<queue>
#include<stack>
#include<random>
#include<bitset>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<unordered_map>
// #pragma GCC optimize(2)
using namespace std;
#define MAXN 905
#define LL long long
#define ULL unsigned long long
#define ENDL putchar('\n')
#define DB double
#define lowbit(x) (-(x) & (x))
#define FI first
#define SE second
#define PR pair<int,int>
#define UIN unsigned int
int xchar() {
	static const int maxn = 1000000;
	static char b[maxn];
	static int pos = 0,len = 0;
	if(pos == len) pos = 0,len = fread(b,1,maxn,stdin);
	if(pos == len) return -1;
	return b[pos ++];
}
// #define getchar() xchar()
LL read() {
	LL f = 1,x = 0;int s = getchar();
	while(s < '0' || s > '9') {if(s<0)return -1;if(s=='-')f=-f;s = getchar();}
	while(s >= '0' && s <= '9') {x = (x<<1) + (x<<3) + (s^48);s = getchar();}
	return f*x;
}
void putpos(LL x) {if(!x)return ;putpos(x/10);putchar((x%10)^48);}
void putnum(LL x) {
	if(!x) {putchar('0');return ;}
	if(x<0) putchar('-'),x = -x;
	return putpos(x);
}
void AIput(LL x,int c) {putnum(x);putchar(c);}

const int MOD = 998244353;
int n,m,s,o,k;
int qkpow(int a,int b) {
	int res = 1;
	while(b > 0) {
		if(b & 1) res = res *1ll* a % MOD;
		a = a *1ll* a % MOD; b >>= 1;
	} return res;
}
int fac[MAXN],inv[MAXN],invf[MAXN];
int A[MAXN];
int dp[35][MAXN],dp2[35][MAXN];
int main() {
	n = read();
	fac[0]=fac[1]=inv[0]=inv[1]=invf[0]=invf[1]=1;
	for(int i = 2;i <= 900;i ++) {
		fac[i] = fac[i-1] *1ll* i % MOD;
		inv[i] = (MOD - inv[MOD%i]) *1ll* (MOD/i) % MOD;
		invf[i] = invf[i-1] *1ll* inv[i] % MOD;
	}
	int sm = 0;
	dp[0][0] = 1;
	for(int i = 1;i <= n;i ++) {
		A[i] = read();
		sm += A[i]-1;
		for(int j = i;j > 0;j --) {
			for(int k = sm;k >= 0;k --) {
				for(int s = 0;s <= k && s < A[i];s ++)
				(dp[j][k] += MOD - dp[j-1][k-s]*1ll*invf[s] % MOD) %= MOD;
			}
		}
	}
	for(int i = 1;i <= n;i ++) {
		memcpy(dp2,dp,sizeof(dp));
		for(int j = 1;j <= n;j ++) {
			for(int k = 0;k <= sm;k ++) {
				for(int s = 0;s <= k && s < A[i];s ++)
				(dp2[j][k] += dp2[j-1][k-s]*1ll*invf[s] % MOD) %= MOD;
			}
		}
		for(int j = n;j > 0;j --) {
			for(int k = A[i]-1;k <= sm;k ++) {
				(dp2[j][k] = dp2[j-1][k-A[i]+1]*1ll*invf[A[i]-1] % MOD) %= MOD;
			}
		}
		int as = 0;
		for(int j = 0;j < n;j ++) {
			int pw = qkpow(inv[n-j],A[i]-1);
			for(int k = A[i]-1;k <= sm;k ++) {
				pw = pw *1ll* inv[n-j] % MOD;
				(as += pw *1ll* dp2[n-j][k] % MOD * fac[k] % MOD) %= MOD;
			}
		}
		AIput(as,i==n ? '\n':' ');
	}
	return 0;
}