POJ 2151 概率DP

主要的子问题是每一个队伍有一个做出题目的概率，求做出k个题目的概率。简单的简单的组合数DP。想清楚即可。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;

double dp[35][35];
double p[1005][35];
int main()
{
//    freopen("1.txt","r",stdin);
    int M,T,N;
    while(cin>>M>>T>>N && M!=0)
    {
        memset(p, 0, sizeof(p));
        memset(dp, 0, sizeof(dp));
        for(int i=0; i<T; i++)
        {
            for(int j=0; j<M; j++)
                cin>>p[i][j];
        }
        double ans = 1.0;
        for(int i=0; i<T; i++)
        {

            double ret = 1.0f;
            for(int j=0; j<M; j++)
                ret *= (1-p[i][j]);
            ret  = 1 - ret;
            ans *= ret;
        }
        double ant = 1.0f;
        for(int i=0; i<T; i++)
        {
            // first i solved num is j
            memset(dp, 0, sizeof(dp));
            dp[1][0] = 1- p[i][0];
            dp[1][1] = p[i][0];
            for(int j=2; j<=M; j++) dp[j][0] = dp[j-1][0] * (1 - p[i][j-1]);
            for(int j=2; j<= M; j++)
            {
                for(int k = 1; k<=j; k++)
                {
                    dp[j][k] = dp[j-1][k-1]*(p[i][j-1])+ dp[j-1][k]*(1 - p[i][j-1]);
                }
            }
            double sum = 0.0f;
            for(int i=1; i<N; i++)
                sum += dp[M][i];
            ant *= sum;
        }
        printf("%.3f
", ans - ant);
    }
    return 0;
}