洛谷P5644 [PKUWC2018] 猎人杀

2021-01-24 10:02:42 阅读：176 来源： 互联网

标签：right 洛谷 int lim sum P5644 frac ll PKUWC2018

有个结论，原问题可以转化为每次开枪的概率中的分母不变，当射到一个已经死掉的猎人时，就继续开枪，不难发现这样射中第 \(i\) 个人的概率和原问题一样，设 \(W=\sum\limits_{i=1}^n w_i\)，\(T\) 为已经死掉的猎人的 \(w_i\) 的和，得：

\[\large\left( \sum_{j=0}^\infty \left(\frac{T}{W}\right)^j\right)\frac{w_i}{W}=\frac{w_i}{W-T} \]

考虑容斥，枚举一个猎人集合 \(S\)，集合内的猎人都要在 \(1\) 之后被射死，得答案为：

\[\large\begin{aligned} &\sum_S(-1)^{|S|}\sum_{i=0}^\infty\left( 1-\frac{\sum\limits_{j\in S}w_j+w_1}{W} \right)\frac{w_1}{W}\\ =&w_1\sum_S\frac{(-1)^{|S|}}{\sum\limits_{j\in S}w_j+w_1}\\ =&w_1\sum_{i=0}^{W-w_1}\frac{\left[x^i\right]\prod\limits_{i=2}^n\left(1-x^{w_i}\right)}{i+w_1}\\ \end{aligned} \]

\(\prod\limits_{i=2}^n\left(1-x^{w_i}\right)\) 用分治和 \(NTT\) 即可计算。

#include<bits/stdc++.h>
#define maxn 400010
#define p 998244353
#define ls (x<<1)
#define rs (x<<1|1)
using namespace std;
typedef long long ll;
template<typename T> inline void read(T &x)
{
    x=0;char c=getchar();bool flag=false;
    while(!isdigit(c)){if(c=='-')flag=true;c=getchar();}
    while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=getchar();}
    if(flag)x=-x;
}
int n,m,w,sum,ans;
ll rev[maxn],inv[maxn],len[maxn],v[maxn],f[maxn];
int mod(int x)
{
    return x>=p?x-p:x;
}
ll qp(ll x,ll y)
{
    ll v=1;
    while(y)
    {
        if(y&1) v=v*x%p;
        x=x*x%p,y>>=1;
    }
    return v;
}
int calc(int n)
{
    int lim=1;
    while(lim<=n) lim<<=1;
    for(int i=0;i<lim;++i)
        rev[i]=(rev[i>>1]>>1)|((i&1)?lim>>1:0);
    return lim;
}
void NTT(ll *a,int lim,int type)
{
    for(int i=0;i<lim;++i)
        if(i<rev[i])
            swap(a[i],a[rev[i]]);
    for(int len=1;len<lim;len<<=1)
    {
        ll wn=qp(3,(p-1)/(len<<1));
        for(int i=0;i<lim;i+=len<<1)
        {
            ll w=1;
            for(int j=i;j<i+len;++j,w=w*wn%p)
            {
                ll x=a[j],y=w*a[j+len]%p;
                a[j]=mod(x+y),a[j+len]=(x-y+p)%p;
            }
        }
    }
    if(type==1) return;
    for(int i=0;i<lim;++i) a[i]=a[i]*inv[lim]%p;
    reverse(a+1,a+lim);
}
void solve(int l,int r,ll *a,int x)
{
    if(l==r)
    {
        for(int i=0;i<=v[l];++i) a[i]=0;
        a[0]=1,a[v[l]]=p-1,len[x]=v[l];
        return;
    }
    int mid=(l+r)>>1,lim;
    ll f[maxn],g[maxn];
    solve(l,mid,f,ls),solve(mid+1,r,g,rs),lim=calc(len[x]=len[ls]+len[rs]);
    for(int i=len[ls]+1;i<lim;++i) f[i]=0;
    for(int i=len[rs]+1;i<lim;++i) g[i]=0;
    NTT(f,lim,1),NTT(g,lim,1);
    for(int i=0;i<lim;++i) a[i]=f[i]*g[i]%p;
    NTT(a,lim,-1);
    for(int i=len[x]+1;i<lim;++i) a[i]=0;
}
int main()
{
    read(n),read(w),sum=w;
    for(int i=1;i<n;++i) read(v[i]),sum+=v[i];
    inv[1]=1;
    for(int i=2;i<=2*sum;++i) inv[i]=(p-p/i)*inv[p%i]%p;
    solve(1,n-1,f,1);
    for(int i=0;i<=sum-w;++i) ans=mod(ans+f[i]*inv[i+w]%p);
    printf("%d",(ll)ans*w%p);
    return 0;
}

标签：right,洛谷,int,lim,sum,P5644,frac,ll,PKUWC2018
来源： https://www.cnblogs.com/lhm-/p/14320128.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

洛谷P5644 [PKUWC2018] 猎人杀