P4564-[CTSC2018]假面【期望dp】
正題
題目大意:https://www.luogu.com.cn/problem/P4564
題目大意
nnn個(gè)人第iii個(gè)有mim_imi?點(diǎn)血,每次有操作
最后要求輸出每個(gè)人的期望血量
解題思路
pi,jp_{i,j}pi,j?表示第iii個(gè)人剩余jjj點(diǎn)血的概率。這個(gè)可以O(Qn)O(Qn)O(Qn)的時(shí)間內(nèi)維護(hù)。
考慮如何計(jì)算概率,因?yàn)榇婊钊藬?shù)的不同,產(chǎn)生的貢獻(xiàn)也不同,我們?cè)O(shè)fi,jf_{i,j}fi,j?表示第iii個(gè)人以外的人存活了jjj個(gè)的概率,這個(gè)很容易可以在O(Cn2)O(Cn^2)O(Cn2)的時(shí)間內(nèi)算,但是這樣顯然過不去。
所以我們要進(jìn)行優(yōu)化,我們可以在O(n2)O(n^2)O(n2)的時(shí)間內(nèi)算出gig_igi?表示所有人里存活iii個(gè)人的概率,也就是有(以下為了方便定義pip_ipi?表示1?pi,01-p_{i,0}1?pi,0?即第iii個(gè)人存活的概率):
gi=gi?1?pu+gi?(1?pu)g_i=g_{i-1}*p_u+g_{i}*(1-p_u)gi?=gi?1??pu?+gi??(1?pu?)
顯然我們可以從fuf_{u}fu?推到ggg
gi=fu,i?(1?pu)+fu,i?1?pug_i=f_{u,i}*(1-p_u)+f_{u,i-1}*p_ugi?=fu,i??(1?pu?)+fu,i?1??pu?可以回推回來也就是
?fu,i=gi?fu,i?1?pu1?pu\Rightarrow f_{u,i}=\frac{g_i-f_{u,i-1}*p_u}{1-p_u}?fu,i?=1?pu?gi??fu,i?1??pu??
這樣我們就可以在O(Cn2)O(Cn^2)O(Cn2)的時(shí)間內(nèi)算出所有的fu,if_{u,i}fu,i?來統(tǒng)計(jì)答案。因?yàn)榍竽嬖埠苈?#xff0c;所以我們先線性推逆元預(yù)處理一下比較小的值。
codecodecode
#include<cstdio> #include<cstring> #include<algorithm> #include<cctype> using namespace std; const int XJQ=998244353; int n,Q,p[210][210],c[210],g[210],f[210],in[210]; int power(int x,int b){int ans=1;x%=XJQ;while(b){if(b&1)ans=1ll*ans*x%XJQ;x=1ll*x*x%XJQ;b>>=1;}return ans; } int read() {int x=0,f=1; char c=getchar();while(!isdigit(c)) {if(c=='-')f=-f;c=getchar();}while(isdigit(c)) x=(x<<1)+(x<<3)+c-48,c=getchar();return x*f; } void print(int x){if (x>9) print(x/10); putchar(x%10+48); return; } signed main() {n=read();for(int i=1;i<=n;i++)p[i][read()]=1;Q=read();in[1]=1;for(int i=2;i<=n;i++)in[i]=(long long)XJQ-(long long)XJQ/i*in[XJQ%i]%XJQ;while(Q--){int op=read();if(op==0){int id=read(),u=read(),v=read();u=1ll*u*power(v,XJQ-2)%XJQ;p[id][0]=(p[id][0]+1ll*p[id][1]*u)%XJQ;for(int i=1;i<=100;i++)p[id][i]=(1ll*p[id][i+1]*u+1ll*p[id][i]*(1-u+XJQ))%XJQ;}else{int k=read(),x,ans=0;for(int i=1;i<=k;i++)c[i]=read();memset(g,0,sizeof(g));g[0]=1;for(int i=1;i<=k;i++){for(int j=i;j>=1;j--)g[j]=(1ll*g[j]*p[c[i]][0]+1ll*(1-p[c[i]][0]+XJQ)*g[j-1])%XJQ;g[0]=1ll*g[0]*p[c[i]][0]%XJQ;}for(int i=1;i<=k;i++){int ans=0,z=(1-p[c[i]][0]+XJQ)%XJQ,inv=power(p[c[i]][0],XJQ-2);if(p[c[i]][0]==1){printf("0 ");continue;}if(p[c[i]][0]==0)for(int j=0;j<k;j++)f[j]=g[j+1];else{f[0]=1ll*g[0]*inv%XJQ;for(int j=1;j<k;j++){f[j]=(g[j]-1ll*f[j-1]*z%XJQ+XJQ)%XJQ;f[j]=1ll*f[j]*inv%XJQ;}}for(int j=0;j<k;j++)(ans+=1ll*f[j]*in[j+1]%XJQ)%=XJQ;print(1ll*ans*z%XJQ);putchar(' ');}putchar('\n'); }}for(int i=1;i<=n;i++){int ans=0;for(int j=1;j<=100;j++)ans=(ans+1ll*p[i][j]*j%XJQ)%XJQ;print(ans);putchar(' ');} }總結(jié)
以上是生活随笔為你收集整理的P4564-[CTSC2018]假面【期望dp】的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: P4552-[Poetize6]IncD
- 下一篇: 恢复数据软件读取不到手机内存卡恢复数据软