【LOJ2541】【PKUWC2018】猎人杀（容斥，FFT）

题面

LOJ

题解

这题好神仙啊。
直接考虑概率很麻烦，因为分母总是在变化。
但是，如果一个人死亡之后，我们不让他离场，假装给他打一个标记（猎人印记？？？）
如果在一次选择的时候选中了一个已经被打过标记的人，那么我们就重新做一次选择。
这样显然没有任何影响。
现在考虑如何求第一个人最后一个被打上标记的概率。
我们容斥一下，枚举一下哪些人会在$1$之后被选择，那么容斥系数就是$(-1)$的人数次方。
那么对于钦定的在$1$之后被选择的集合$S$，假设他们的$w$的和为$S$，所有人$w$的和为$A$。这个集合贡献的值就是$(-1)^{|S|}\sum_{i=1}^{\infty}(1-\frac{S+W_1}{A})^i\frac{W_1}{A}$。后面的部分可以化简，结果就是$\frac{W_1}{S+W_1}$。
现在的问题就是求$S$了。
我们发现因为是一个分数的形式，如果直接计算显然是不能够直接$dp$。
换种思路，如果我们计算每一种分母分别出现了多少次，这个就非常好算了。
这样就可以通过$dp$计算，同时发现事实上这个$dp$就是一个$01$背包，所以可以用分治+$NTT$来优化计算。
时间复杂度$O(nlogn^2)$

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
using namespace std;
#define MAX 250000
#define ll long long
#define MOD 998244353
inline int read()
{
    int x=0;bool t=false;char ch=getchar();
    while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
    if(ch=='-')t=true,ch=getchar();
    while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
    return t?-x:x;
}
int fpow(int a,int b)
{
    int s=1;
    while(b){if(b&1)s=1ll*s*a%MOD;a=1ll*a*a%MOD;b>>=1;}
    return s;
}
int n,w[MAX],ans;
int W[MAX],r[MAX];
void NTT(int *P,int len,int opt)
{
    int N,l=0;for(N=1;N<len;N<<=1)++l;
    for(int i=0;i<N;++i)r[i]=(r[i>>1]>>1)|((i&1)<<(l-1));
    for(int i=0;i<N;++i)if(i<r[i])swap(P[i],P[r[i]]);
    for(int i=1;i<N;i<<=1)
    {
        int w=fpow(3,(MOD-1)/(i<<1));W[0]=1;
        for(int k=1;k<i;++k)W[k]=1ll*W[k-1]*w%MOD;
        for(int p=i<<1,j=0;j<N;j+=p)
            for(int k=0;k<i;++k)
            {
                int X=P[j+k],Y=1ll*P[i+j+k]*W[k]%MOD;
                P[j+k]=(X+Y)%MOD;P[i+j+k]=(X+MOD-Y)%MOD;
            }
    }
    if(opt==-1)
    {
        reverse(&P[1],&P[N]);
        for(int i=0,inv=fpow(N,MOD-2);i<N;++i)P[i]=1ll*P[i]*inv%MOD;
    }
}
int tmp[50][MAX];
int S[MAX],top,P[MAX];
int Solve(int l,int r,int *P)
{
    if(l==r){P[0]=1;P[w[l]]=MOD-1;return w[l];}
    int mid=(l+r)>>1,l1,l2,ls,rs,N;
    ls=S[top--];l1=Solve(l,mid,tmp[ls]);
    rs=S[top--];l2=Solve(mid+1,r,tmp[rs]);
    for(N=1;N<=l1+l2;N<<=1);
    NTT(tmp[ls],N,1);NTT(tmp[rs],N,1);
    for(int i=0;i<N;++i)P[i]=1ll*tmp[ls][i]*tmp[rs][i]%MOD;
    NTT(P,N,-1);S[++top]=ls;S[++top]=rs;
    for(int i=0;i<N;++i)tmp[ls][i]=tmp[rs][i]=0;
    return l1+l2;
}
int main()
{
    n=read();
    for(int i=1;i<=n;++i)w[i]=read();
    for(int i=0;i<50;++i)S[++top]=i;
    int len=Solve(2,n,P);
    for(int i=0;i<=len;++i)ans=(ans+1ll*P[i]*fpow(w[1]+i,MOD-2))%MOD;
    ans=1ll*ans*w[1]%MOD;
    printf("%d\n",ans);
    return 0;
}