bzoj 3029 守衛者的挑戰——概率期望dp+狀態數思考
阿新 • • 發佈:2018-07-25
www. 發現 targe get 有關 lan clu else print
題目:https://www.lydsy.com/JudgeOnline/problem.php?id=3029
先隨便寫了個dfs,記錄“前 i 次、成功 j 次、容量-殘片=k”的概率。因為是否可行只和“成功次數”還有“容量-殘片個數”有關,和容量、殘片具體數量無關。準備記憶化,但發現狀態存不下。
#include<iostream> #include<cstdio> #include<cstring> #define db double using namespace std; const int N=205,M=202005; int n,l,K,a[N]; db p[N],ans;View Codevoid dfs(int i,int j,int c,double w)//至少有c個 { if(c>n)c=n; if(i>n) { if(j>=l&&c>=0)ans+=w;return; } if(j+(n-i+1)<l)return; if(a[i]>=0) { dfs(i+1,j+1,c+a[i],w*p[i]); dfs(i+1,j,c,w*(1-p[i])); } else { dfs(i+1,j+1,c-1,w*p[i]); dfs(i+1,j,c,w*(1-p[i])); } } int main() { scanf("%d%d%d",&n,&l,&K);int x; for(int i=1;i<=n;i++) { scanf("%d",&x);p[i]=(db)x/100; } for(int i=1;i<=n;i++)scanf("%d",&a[i]); dfs(1,0,K,1); printf("%.6lf\n",ans);return 0; }
然後想到了 bzoj4247 掛飾 的思路。就是發現“容量-殘片數量”大於n的話多出來的部分沒什麽用,可以把大於n記成等於n。這樣第三維就只有2*n啦!然後就可以刷表了。
#include<iostream> #include<cstdio> #include<cstring> #define db double using namespace std; const int N=205,M=202005; int n,l,K,a[N]; db p[N],ans,dp[2][N][N<<1]; int main() { scanf("%d%d%d",&n,&l,&K);int x; for(int i=1;i<=n;i++) { scanf("%d",&x);p[i]=(db)x/100; } for(int i=1;i<=n;i++)scanf("%d",&a[i]); dp[0][0][min(n,K)+N]=1; for(int i=0;i<n;i++) { int d=(i&1); for(int j=0;j<i;j++) for(int k=-n;k<=n;k++)dp[!d][j][k+N]=0; for(int j=0;j<=i;j++) for(int k=-n;k<=n;k++) { if(a[i+1]>=0) { dp[!d][j+1][min(n,k+a[i+1])+N]+=dp[d][j][k+N]*p[i+1];//+= dp[!d][j][k+N]+=dp[d][j][k+N]*(1-p[i+1]); } else { dp[!d][j+1][k-1+N]+=dp[d][j][k+N]*p[i+1]; dp[!d][j][k+N]+=dp[d][j][k+N]*(1-p[i+1]); } } } int d=(n&1); for(int j=l;j<=n;j++) for(int k=0;k<=n;k++) ans+=dp[d][j][k+N]; printf("%.6lf\n",ans); return 0; }
bzoj 3029 守衛者的挑戰——概率期望dp+狀態數思考