「PKUWC2018」LOJ #2541. 猎人杀 - 容斥 - 分治NTT

本文探讨了一个涉及球权重与扔球顺序概率的问题,通过容斥原理和分治NTT技术,提出了一种新颖的算法解决方案,用于计算在特定条件下最后一号球被扔出的概率。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目大意:给你nnn个球,每个球有个权重wiw_iwi,每次加权扔掉一个球,问最后一号球是最后一个被扔出去的概率。∑i=1nwi≤105,∀i∈[1,n],wi>0\sum_{i=1}^n w_i\le10^5,\forall i\in[1,n],w_i>0i=1nwi105,i[1,n],wi>0
题解:
首先考虑容斥(本质上是Min−MaxMin-MaxMinMax容斥,即要求其为全集的最大值,可以通过要求其为全集的子集的最小值来容斥),容斥至少SSS这个集合中的球在111号之后被扔掉的概率α(S)\alpha(S)α(S),那么答案是
∑S(−1)∣S∣α(S)\sum_{S}(-1)^{|S|}\alpha(S)S(1)Sα(S)
考虑α(S)\alpha(S)α(S)怎么算,我们可以如此转化:每次随机选择一个球,若其被选择过了,就忽略不计。可以感性理解这样做和原来选择一个就扔掉一个是本质相同的。那么要求111SSS之前出现,等价于,若枚举其是第i+1i+1i+1个出现的,则前面选择(注意这里不是出现)的都不是S⋃{1}S\bigcup\{1\}S{1}的元素,即:
α(S)=∑i≥0(1−A+w1W)iw1W=WA+w1w1W=w1A+w1\alpha(S)=\sum_{i\geq0}\left(1-\frac{A+w_1}{W}\right)^i\frac{w_1}{W}=\frac{W}{A+w_1}\frac{w_1}{W}=\frac{w_1}{A+w_1}α(S)=i0(1WA+w1)iWw1=A+w1WWw1=A+w1w1
其中W=∑i=1nwi,  A=∑x∈SwxW=\sum_{i=1}^nw_i,\ \ A=\sum_{x\in S}w_xW=i=1nwi,  A=xSwx
注意到W≤105W\le10^5W105,使用分治NTT计算AAA固定时的容斥系数即可。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#define gc getchar()
#define pb push_back
#define lint long long
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define clr(a,n) memset(a,0,sizeof(int)*(n))
#define p 998244353
#define N 100010
#define debug(x) cerr<<#x<<"="<<x
#define sp <<" "
#define ln <<endl
using namespace std;
inline int inn()
{
	int x,ch;while((ch=gc)<'0'||ch>'9');
	x=ch^'0';while((ch=gc)>='0'&&ch<='9')
		x=(x<<1)+(x<<3)+(ch^'0');return x;
}
vector<int> v[N];int w[N],r[N<<2],a[N<<2],b[N<<2];
inline int fast_pow(int x,int k,int ans=1)
{	for(x%=p;k;k>>=1,x=(lint)x*x%p) (k&1)?ans=(lint)ans*x%p:0;return ans;	}
inline int NTT(int *a,int n,int s)
{
	for(int i=1;i<n;i++) if(i<r[i]) swap(a[i],a[r[i]]);
	for(int i=2;i<=n;i<<=1)
	{
		int wn=fast_pow(3,s>0?(p-1)/i:p-1-(p-1)/i);
		for(int j=0,t=i>>1;j<n;j+=i) for(int k=0,w=1,x,y;k<t;k++,w=(lint)w*wn%p)
			x=a[j+k],y=(lint)w*a[j+k+t]%p,((a[j+k]=x+y)>=p?a[j+k]-=p:0),((a[j+k+t]=x-y)<0?a[j+k+t]+=p:0);
	}
	if(s<0) for(int i=0,ninv=fast_pow(n,p-2);i<n;i++) a[i]=(lint)a[i]*ninv%p;return 0;
}
inline int tms(vector<int> &A,vector<int> &B,vector<int> &C)
{
	int m1=(int)A.size(),m2=(int)B.size(),n=1,L=0;
	while(n<m1+m2-1) n<<=1,L++;
	rep(i,1,n-1) r[i]=(r[i>>1]>>1)|((i&1)<<(L-1));
	rep(i,0,m1-1) a[i]=A[i];clr(a+m1,n-m1),NTT(a,n,1);
	rep(i,0,m2-1) b[i]=B[i];clr(b+m2,n-m2),NTT(b,n,1);
	rep(i,0,n-1) a[i]=(lint)a[i]*b[i]%p;NTT(a,n,-1);
	C.resize(m1+m2-1);rep(i,0,m1+m2-2) C[i]=a[i];return 0;
}
inline int solve(int l,int r)
{
	if(l==r) return v[l].resize(w[l]+1,0),v[l][0]=1,v[l][w[l]]=p-1;int mid;
	return mid=(l+r)>>1,solve(l,mid),solve(mid+1,r),tms(v[l],v[mid+1],v[l]);
}
int main()
{
	int n=inn(),s=0;lint ans=0;
	if(n==1) return !printf("1\n");
	rep(i,1,n) w[i]=inn(),s+=(i>1)*w[i];solve(2,n);
	rep(i,0,s) ans+=(lint)v[2][i]*fast_pow(i+w[1],p-2)%p;
	return !printf("%lld\n",ans%p*w[1]%p);
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值