[CTSC2018]假面 01背包的删除

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/DOFYPXY/article/details/80334610

首先记录ai,j表示单位i生命值为j的概率,那么每次修改可以O(m)
对于询问,先求出exi=j=1miai,j表示单位i存活的概率。那么我们只要对于每一个单位i,求出除了该单位的剩下k1个单位中,有1..k1个单位的存活的概率就解决了。
fi,j表示考虑前i个单位,其中j个单位存活的概率,转移一轮是O(n2)的,因为对于每个单位都要转移一轮,所以是O(n3)
因为每轮转移都只有自己没有参与转移,考虑如何从01背包中删除一个元素。
添加一个元素fk=fk1exi+fk(1exi)
那么删除一个元素就是fk=fkfk1exi1exi
只需要先把所有的转移一遍,再分别删除每一个即可。注意处理exi=1的情况,总复杂度O(Qm+Cn2)
代码:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 210
#define M 110
#define ll long long
using namespace std;
const int mod=998244353;
int n,c[N],Q,t[N];
ll a[N][M],f[N][N],g[N],ex[N],ans[N],inv[N];
ll ksm(ll a,ll b){ll r=1;for(;b;b>>=1){if(b&1)r=r*a%mod;a=a*a%mod;}return r;}


int read()
{
    int x=0,f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar()) if(ch=='-') f=-1;
    for(;ch>='0'&&ch<='9';ch=getchar()) x=x*10+ch-'0';
    return x*f;
}
int main()
{   
    n=read();
    for(int i=0;i<=n+1;i++)
        inv[i]=ksm(i,mod-2);
    for(int i=1;i<=n;i++)
        c[i]=read(),a[i][c[i]]=1;
    Q=read();
    while(Q--)
    {
        int opt=read();
        if(opt)
        {
            int m=read();

            memset(f,0,sizeof(f));
            memset(g,0,sizeof(g));
            memset(ex,0,sizeof(ex));
            for(int i=1;i<=m;i++)
                t[i]=read();
            for(int i=1;i<=m;i++)
                ex[i]=(mod+1-a[t[i]][0])%mod;   
            f[0][0]=1;
            for(int i=1;i<=m;i++)
                for(int j=0;j<=i;j++)
                    f[i][j]=((j>0?f[i-1][j-1]*ex[i]:0)+f[i-1][j]*(mod+1-ex[i]))%mod;                    
            for(int i=1;i<=m;i++)
                ans[i]=0;       
            for(int i=1;i<=m;i++)
                if(ex[i])
                {
                    if(ex[i]==1) 
                        for(int j=1;j<=m;j++)
                            g[j-1]=f[m][j];
                    else
                    {
                        ll tmp=ksm(mod+1-ex[i],mod-2);
                        g[0]=f[m][0]*tmp%mod;
                        for(int j=1;j<=m;j++)
                            g[j]=(f[m][j]-g[j-1]*ex[i]%mod+mod)*tmp%mod;
                    }
                    for(int j=1;j<=m;j++)
                        ans[i]=(ans[i]+ex[i]*g[j-1]%mod*inv[j])%mod;
                } 
            for(int i=1;i<=m;i++)           
                printf("%lld ",ans[i]);
            puts("");   
        }
        else 
        {
            int id=read();ll u=read(),v=read(),P=u*ksm(v,mod-2)%mod;
            a[id][0]=(a[id][0]+P*a[id][1])%mod;
            for(int i=1;i<=c[id];i++)
                a[id][i]=(a[id][i]*(mod+1-P)+a[id][i+1]*P)%mod;
        }
    }
    for(int i=1;i<=n;i++)
        ans[i]=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=c[i];j++)
            ans[i]=(ans[i]+a[i][j]*j)%mod;
    for(int i=1;i<=n;i++)
        printf("%lld ",ans[i]);     
    puts("");       
    return 0;
}
阅读更多
想对作者说点什么?
相关热词

博主推荐

换一批

没有更多推荐了,返回首页