P5644 [PKUWC2018]猎人杀

P5644 [PKUWC2018]猎人杀

题目大意

一开始有$n$个猎人，第$i$个猎人有仇恨度$w_i$。每次可以开枪射杀一个活着的猎人。

假设活着的猎人为$i_1,i_2,\dots ,i_m$，则第$i_k$个猎人被射杀的概率是$\frac{w_{i_k}}{\sum _{j=1}^m{w}_{i_j}}$。

求$1$号猎人最后一个被射杀的概率。输出答案对$998244353$取模。

$w_i>0,1\le \sum w_i\le 10^5$

题解

转化题意

在不断射杀的过程中，概率的分母会不断改变。所以，我们可以稍微转化一下题意。

令$sum=\sum _{i=1}^n{w}_i$，题意转化为：第$i$个人被射杀的概率为$\frac{w_i}{sum}$，已经被射杀的人仍能继续被射杀。如果打中一个活着的人，那么这个人就死去。

为什么呢？因为在每次射杀的时候，每个活人被射杀的概率都他的仇恨度除以所有活人仇恨度的和。这样相对于原来，多了一个射杀死人的过程，但因为活人最终总会被射杀，所以本质上是一样的。

容斥

直接求$1$号最后被射杀的概率比较困难，我们考虑使用容斥。

设在$1$号之后被射杀的人的子集为$S$的概率为$p(S)$，则答案为

$$ans=\sum (-1{)}^{|S|}p(S)$$

然后，我们考虑如何求$p(S)$。

设$v(S)=\sum _{i\in S}{w}_i$。因为在$1$号之后被射杀的人包含$S$，所以就相当于射杀若干次，每次射杀除了$1$号和集合$S$之外的人，直到打中$1$号。

$$p(S)=\sum _{i=0}^{+\infty }(\frac{sum-w_1-v(S)}{sum}{)}^i\cdot \frac{w_1}{sum}$$

接下来求$\sum _{i=0}^{+\infty }(\frac{sum-w_1-v(S)}{sum}{)}^i$。由等比数列求和公式可得

$$\sum _{i=0}^{+\infty }(\frac{sum-w_1-v(S)}{sum}{)}^i=\frac{1-(\frac{sum-w_1-v(S)}{sum}{)}^{+\infty }}{1-\frac{sum-w_1-v(S)}{sum}}=\frac{1}{\frac{w_1+v(S)}{sum}}=\frac{sum}{w_1+v(S)}$$

所以

$$p(S)=\frac{sum}{w_1+v(S)}\cdot \frac{w_1}{sum}=\frac{w_1}{w_1+v(S)}$$

那么

$$ans=\sum (-1{)}^{|S|}\frac{w_1}{w_1+v(S)}$$

生成函数

依题意，$1\le \sum w_i\le 10^5$，所以我们可以枚举$v(S)$。

令

$$g(i)=\sum _{v(S)=i}(-1{)}^{|S|}$$

那么

$$ans=\sum _{i=0}^{sum}g(i)\cdot \frac{w_1}{w_1+i}$$

于是问题就转化为求$g(i)$了。我们发现，$g(i)$其实就是$\prod _{i=2}^n(1-x^{w_i})$的第$i$次项的系数。

$NTT$

接下来，我们要用$NTT$来求$\prod _{i=2}^n(1-x^{w_i})$。

用分治，求$[l,r]$的多项式时，先求出$[l,mid]$和$[mid+1,r]$的多项式，再将两个多项式相乘即可。

我们把求的过程看作$\log n$层，每层的时间复杂度为$O(sum\log sum)$，所以总时间复杂度为$O(sum\log sum\log n)$。

总时间复杂度可以看作$O(n{\log }^{2}n)$。

code

#include<bits/stdc++.h>
using namespace std;
long long ans=0,w[100005],f[500005],g[20][500005];
const long long G=3,mod=998244353;
long long mi(long long t,long long v){
	if(!v) return 1;
	long long re=mi(t,v/2);
	re=re*re%mod;
	if(v&1) re=re*t%mod;
	return re;
}
void ch(long long *a1,int l){
	for(int i=1,j=l/2;i<l-1;i++){
		if(i<j) swap(a1[i],a1[j]);
		int k=l/2;
		while(j>=k){
			j-=k;k>>=1;
		}
		j+=k;
	}
}
void ntt(long long *a1,int l,int fl){
	long long W,wn;
	for(int i=2;i<=l;i<<=1){
		if(fl==1) wn=mi(G,(mod-1)/i);
		else wn=mi(G,mod-1-(mod-1)/i);
		for(int j=0;j<l;j+=i){
			W=1;
			for(int k=j;k<j+i/2;k++,W=W*wn%mod){
				long long t=a1[k],u=W*a1[k+i/2]%mod;
				a1[k]=(t+u)%mod;
				a1[k+i/2]=(t-u+mod)%mod;
			}
		}
	}
	if(fl==-1){
		long long ny=mi(l,mod-2);
		for(int i=0;i<l;i++) a1[i]=a1[i]*ny%mod;
	}
}
int solve(int l,int r,long long *a1,int now){
	if(l==r){
		a1[0]=1;a1[w[l]]=mod-1;
		return w[l];
	}
	int mid=l+r>>1,vt,len=1;
	vt=solve(l,mid,a1,now+1)+solve(mid+1,r,g[now+1],now+1);
	while(len<=vt) len<<=1;
	ch(a1,len);ch(g[now+1],len);
	ntt(a1,len,1);ntt(g[now+1],len,1);
	for(int i=0;i<len;i++){
		a1[i]=a1[i]*g[now+1][i]%mod;
		g[now+1][i]=0;
	}
	ch(a1,len);
	ntt(a1,len,-1);
	return vt;
}
int main()
{
	int n;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%lld",&w[i]);
	}
	if(n==1){
		printf("1");return 0;
	}
	int vt=solve(2,n,f,0);
	for(int i=0;i<=vt;i++){
		ans=(ans+f[i]*w[1]%mod*mi(w[1]+i,mod-2)%mod)%mod;
	}
	printf("%lld",ans);
	return 0;
}