POJ 2151 Check the difficulty of problems (概率DP)

题目地址：POJ 2151

dp[i][j][k]表示第i个人前j个题做对了k道题的概率。然后把dp数组求出来之后，再利用容斥原理求出最终概率。

代码如下;

#include <iostream>
#include <string.h>
#include <math.h>
#include <queue>
#include <algorithm>
#include <stdlib.h>
#include <map>
#include <set>
#include <stdio.h>
using namespace std;
#define LL __int64
#define pi acos(-1.0)
const int mod=100000000;
const int INF=0x3f3f3f3f;
const double eqs=1e-8;
double dp[1002][40][40];
double p[1002][40];
int main()
{
        int n, m, t, i, j, k;
        double ans1, ans2, p1, p2;
        while(scanf("%d%d%d",&m,&t,&n)!=EOF&&(m||t||n)) {
                for(i=1; i<=t; i++) {
                        for(j=1; j<=m; j++) {
                                scanf("%lf",&p[i][j]);
                        }
                }
                memset(dp,0,sizeof(dp));
                for(i=1; i<=t; i++) {
                        dp[i][1][0]=1-p[i][1];
                        dp[i][1][1]=p[i][1];
                }
                for(i=1; i<=t; i++) {
                        for(j=2; j<=m; j++) {
                                for(k=0; k<=j; k++) {
                                        dp[i][j][k]+=dp[i][j-1][k]*(1-p[i][j]);
                                        if(k)
                                                dp[i][j][k]+=dp[i][j-1][k-1]*p[i][j];
                                }
                        }
                }
                ans1=ans2=1;
                for(i=1; i<=t; i++) {
                        p1=p2=0;
                        for(j=1; j<=m; j++) {
                                p1+=dp[i][m][j];
                                if(j>=1&&j<n)
                                        p2+=dp[i][m][j];
                        }
                        ans1*=p1;
                        ans2*=p2;
                }
                printf("%.3f\n",ans1-ans2);
        }
        return 0;
}