【UOJ #390】【UNR #3】百鸽笼

Description

给定$n$个正整数$a_i$，令$N+1=\sum a_i$
将执行$N$次操作，每次等概率随机选择一个非零的$a_i$并令其减一，显然$N$次操作结束之后有且仅有一个$a_i=1$
对于一开始的$n$个$a_i$，分别求出它们最后为$1$的概率
$n,a_i\le 30$

Analysis

写这题时作死没开longlong，结果还是被乘法忘记转longlong坑死了
以后这种题无脑上longlong，千万别玩火
每次选择非零$a_i$减一，每次操作概率会更改，不好统计答案。
考虑将操作转化成，随机选择$a_i$，若$a_i>0$则令其减一，否则继续随机，显然答案等价。这样每次操作的每个$a_i$被选中的概率都是$1/n$
使用EGF来描述操作序列，为了方便计算，我们把最后一个剩下的数也加入操作序列。不妨假设最后剩下的是$a_1$
对于$2\le i\le n$，$i$至少出现$a_i$次；对于$1$，$1$恰好出现$a_1$次，且一定在序列末出现
令$G_k(x)={\sum }_{i=0}^{a_k-1}\frac{x^i}{i!}$，那么合法序列（除最后一项）的EGF就是
$$F(x)=\frac{x^{a_1-1}}{(a_1-1)!}\prod _{i̸=1}(e^x-G_i(x))$$
答案即为
$$\sum _{i\ge 0}\frac{i![x^i]F(x)}{n^{i+1}}$$
将$F(x)$展开，每部分形如$\lambda x^d{e}^{jx}$，其中$\lambda$可以$dp$求出，考虑计算该部分对答案的贡献
$$\frac{\lambda }{n^{d+1}}\sum _{i\ge 0}\frac{j^i(d+i)!}{i!n^i}$$
$$=\frac{\lambda d!}{n^{d+1}}\sum _{i\ge 0}(\frac{j}{n}{)}^i(\genfrac{}{}{0ex}{}{d+i}{i})$$
$$=\frac{\lambda d!}{n^{d+1}}(\frac{n}{n-j}{)}^{d+1}$$
最后还剩下一个问题，如果对于每个$a_i$我们都暴力做dp来求$\lambda$，令 m = m a x { a i } m=max\{a_i\} m=max{ai​}复杂度就是$O(n^4{m}^2)$，会TLE
正确姿势是一开始求出${\prod }_i(e^x-G_i(x))$，对于每个$a_i$再消除它
复杂度$O(n^3{m}^2)$

Code

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<assert.h>
#define fo(i,a,b) for(int i=(a);i<=(b);++i)
#define fd(i,b,a) for(int i=(b);i>=(a);--i)
using namespace std;
typedef long long ll;
int read(){int n=0,p=1;char ch;for(ch=getchar();ch<'0' || ch>'9';ch=getchar())if(ch=='-') p=-1;for(;'0'<=ch && ch<='9';ch=getchar()) n=n*10+ch-'0';return n*p;}
const int N=33,mo=998244353;
void inc(int &x,int y){x=(x+y)%mo;}
int qmi(int x,int n)
{
	int t=1;
	for(x%=mo;n;n>>=1,x=1ll*x*x%mo) if(n&1) t=1ll*t*x%mo;
	return t;
}
int n,m,a[N],f[N][N][N*N],g[N][N*N],fac[N*N],ifac[N*N],inv[N*N];
int main()
{
	n=read();
	fo(i,1,n) a[i]=read(),m=max(m,a[i]);
	fac[0]=ifac[0]=inv[1]=1;
	fo(i,1,n*m) fac[i]=1ll*fac[i-1]*i%mo;
	ifac[n*m]=qmi(fac[n*m],mo-2);
	fd(i,n*m-1,1) ifac[i]=1ll*ifac[i+1]*(i+1)%mo,inv[i+1]=1ll*ifac[i+1]*fac[i]%mo;
	
	f[0][0][0]=1;
	fo(i,1,n)
		fo(j,0,i)
			fo(k,0,i*m)
			{
				if(j) inc(f[i][j][k],f[i-1][j-1][k]);
				fo(l,0,min(a[i]-1,k))
					inc(f[i][j][k],-1ll*f[i-1][j][k-l]*ifac[l]%mo);
			}
	fo(i,1,n)
	{
		memset(g,0,sizeof(g));
		fd(j,n-1,0)
			fo(k,0,(n-1)*m)
			{
				g[j][k]=f[n][j+1][k];
				fo(l,0,min(a[i]-1,k))
					inc(g[j][k],1ll*g[j+1][k-l]*ifac[l]%mo);
			}
		int ans=0;
		fo(j,0,n-1)
			fo(k,0,(n-1)*m) if(g[j][k])
			{
				int lamda=1ll*ifac[a[i]-1]*g[j][k]%mo,d=(a[i]-1)+k;
				int t=1ll*lamda*qmi(inv[n-j],d+1)%mo*fac[d]%mo;
				ans=(ans+t)%mo;
			}
		printf("%d ",(ans%mo+mo)%mo);
	}
	return 0;
}