概率 DP POJ 2151

蛮简单的一道概率 DP，

有 t 个队，答 m 道题目，给出每个队伍答对每道题的概率

求 所有队均答对一道题并且冠军答对不小于 n 题的概率

冠军答对超过 n 等价于 存在一个队伍答题不小于 n ，对立事件 所有队伍答题少于 n（当然，在所有队伍答题不少于1的条件下）

dp[i][j][k] 代表第i队，前 j 道题目 答对了K 道。

我觉得 DP 还是表示状态很重要，表示清楚状态了，转移就顺理成章的表示出来了。

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <string.h>

using namespace std;

double dp[1111][40][40];// i team before j problem solve k  

double num[1111][40];
int main()
{
    int m,t,n;
    freopen("acm.in","r",stdin);
    while(scanf("%d%d%d",&m,&t,&n)&&m!=0)
    {
        memset(dp,0,sizeof(dp));
        for(int i=1;i<=t;i++)
            for(int j=1;j<=m;j++)
                scanf("%lf",&num[i][j]);
        for(int i=1;i<=t;i++)
            dp[i][0][0]=1;
        for(int i=1;i<=t;i++)
        {
            for(int j=1;j<=m;j++)
            {
                for(int k=0;k<=j;k++)
                {
                    if(k==0)
                        dp[i][j][0]=dp[i][j-1][0]*(1-num[i][j]);
                    else
                        dp[i][j][k]=dp[i][j-1][k-1]*num[i][j]+dp[i][j-1][k]*(1-num[i][j]);
                }
            }
        }
        double res=0;
        double least=1.0;
        for(int i=1;i<=t;i++)
            least*=(1-dp[i][m][0]);
        double mid=1.0;
        for(int i=1;i<=t;i++)
        {
            double tmp=0;
            for(int j=1;j<n;j++)
                tmp+=dp[i][m][j];
            mid*=tmp;
        }
        printf("%.3f\n",least-mid);
    }
    return 0;
}