Check the difficulty of problems - poj 2151 (概率+DP)

 1 #include<stdio.h>
 2 #include<stdlib.h>
 3 #include<string.h>
 4 int M,T,N;
 5 double p[1002][32];
 6 double dp[1002][32][32];
 7 double sum[1002][32];
 8 int main() {
 9     scanf("%d %d %d",&M,&T,&N);
10     while(M&&N&&T){
11         memset(p,0,sizeof(p));
12         for(int i=1;i<=T;i++){
13             for(int j=1;j<=M;j++){
14                 scanf("%lf",&p[i][j]);
15             }
16         }
17         memset(dp,0,sizeof(dp));
18         for(int i=1;i<=T;i++){
19             dp[i][1][1]=p[i][1];
20             dp[i][1][0]=1-p[i][1];
21 
22         }
23         for(int i=1;i<=T;i++){
24             for(int j=2;j<=M;j++){
25                 for(int k=0;k<=j;k++){
26                     dp[i][j][k]=dp[i][j-1][k]*(1.0-p[i][j]);
27                     if(k>0) dp[i][j][k]+=dp[i][j-1][k-1]*p[i][j];
28                 }
29             }
30         }
31         memset(sum,0,sizeof(sum));
32         for(int i=1;i<=T;i++){
33              sum[i][0]=dp[i][M][0];
34             for (int j=1;j<=M;j++) {
35                    sum[i][j]=sum[i][j-1]+dp[i][M][j];
36             }
37         }
38         double ans1=1.0,ans2=1.0;
39         for(int i=1; i<=T; i++)  {
40                ans1*=sum[i][M]-sum[i][0];
41                ans2*=(sum[i][N-1]-sum[i][0]);
42             }
43         printf("%.3lf\n",ans1-ans2);
44         scanf("%d %d %d",&M,&T,&N);
45     }
46 
47     return 0;
48 }