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

最新推荐文章于 2022-06-19 09:52:21 发布

cervoliu

最新推荐文章于 2022-06-19 09:52:21 发布

阅读量780

点赞数

CC 4.0 BY-SA版权

分类专栏：题解数学文章标签： EGF

本文链接：https://blog.csdn.net/lyd_7_29/article/details/83787557

题解同时被 2 个专栏收录

287 篇文章

订阅专栏

数学

37 篇文章

订阅专栏

本文探讨了如何利用指数生成函数(EGF)解决一类特定的概率问题，即在给定一系列正整数并进行特定操作后，计算每个数最终成为1的概率。通过巧妙转换操作，将问题转化为EGF的形式，并详细解析了计算过程，包括DP方法求解参数λ，以及如何优化复杂度避免TLE。

Description

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

Analysis

~~写这题时作死没开longlong，结果还是被乘法忘记转longlong坑死了~~
~~以后这种题无脑上longlong，千万别玩火~~
每次选择非零 $a_i$ 减一，每次操作概率会更改，不好统计答案。
考虑将操作转化成，随机选择 $a_i$ ，若 $a_i>0$ 则令其减一，否则继续随机，显然答案等价。这样每次操作的每个 $a_i$ 被选中的概率都是 $1 / n$
使用EGF来描述操作序列，为了方便计算，我们把最后一个剩下的数也加入操作序列。不妨假设最后剩下的是 $a_1$
对于 $2≤i≤n2\leq i\leq n$ ， $i$ 至少出现 $a_i$ 次；对于 $1$ ， $1$ 恰好出现 $a_1$ 次，且一定在序列末出现
令 $Gk(x)=∑i=0ak−1xii!G_k(x)=\sum_{i=0}^{a_k-1}\frac{x^i}{i!}$ ，那么合法序列（除最后一项）的EGF就是
$F(x)=xa1−1(a1−1)!∏i≠1(ex−Gi(x))F(x)=\frac{x^{a_1-1}}{(a_1-1)!}\prod_{i\neq 1}(e^x-G_i(x))$
答案即为
$∑i≥0i![xi]F(x)ni+1\sum_{i\geq 0}\frac{i![x^i]F(x)}{n^{i+1}}$
将 $F (x)$ 展开，每部分形如 $λxdejx\lambda x^d e^{jx}$ ，其中 $λ\lambda$ 可以 $d p$ 求出，考虑计算该部分对答案的贡献
$λnd+1∑i≥0ji(d+i)!i!ni\frac{\lambda}{n^{d+1}} \sum_{i\geq 0}\frac{j^i (d+i)!}{i!n^i}$
$=λd!nd+1∑i≥0(jn)i(d+ii)=\frac{\lambda d!}{n^{d+1}} \sum_{i\geq 0}(\frac{j}{n})^i {d+i \choose i}$
$=λd!nd+1(nn−j)d+1=\frac{\lambda d!}{n^{d+1}} (\frac{n}{n-j})^{d+1}$
最后还剩下一个问题，如果对于每个 $a_i$ 我们都暴力做dp来求 $λ\lambda$ ，令 $m=max\{a_i\}$ 复杂度就是 $O(n^4 m^2)$ ，会TLE
正确姿势是一开始求出 $∏i(ex−Gi(x))\prod_{i}(e^x-G_i(x))$ ，对于每个 $a_i$ 再消除它
复杂度 $O(n^3m^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;
}