1. 程式人生 > >bzoj 3029: 守衛者的挑戰 概率dp

bzoj 3029: 守衛者的挑戰 概率dp

題意

有n個位置,每個位置有一個權值ai,若ai<0則ai必為-1,且有pi的概率獲得這個值。初始值為K。現在問至少獲得了L個值且獲得的值的和不小於0的概率。
n<=200,L<=n,ai,K<=2000

分析

設f[i,j,k]表示到第i個位置,拿了j個值,當前和為k的概率。
顯然若當前的值大於n的話就一定可以滿足和不小於0,那麼就把大於n的放到n這裡就好了。
轉移隨便yy一下就好了。

程式碼

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm> #include<cmath> using namespace std; const int N=201; int n,m,L,a[N]; double p[N],f[N][N][401]; void modify(int i,int j,int k,double w) { k=min(k,n); f[i][j][k+n]+=w; } int main() { scanf("%d%d%d",&n,&L,&m); for (int i=1;i<=n;i++) scanf
("%lf",&p[i]),p[i]=1.0*p[i]/100; for (int i=1;i<=n;i++) scanf("%d",&a[i]); if (m>n) f[0][0][n+n]=1; else f[0][0][m+n]=1; for (int i=0;i<n;i++) for (int j=0;j<=i;j++) for (int k=-n;k<=n;k++) { modify(i+1,j,k,1.0*(1-p[i+1])*f[i][j][k+n]); modify(i+1
,j+1,k+a[i+1],1.0*p[i+1]*f[i][j][k+n]); } double ans=0; for (int j=L;j<=n;j++) for (int k=0;k<=n;k++) ans+=f[n][j][k+n]; printf("%.6lf",ans); return 0; }