POJ 2151 Check the difficulty of problems(概率dp啊)

标签：dp   poj   

题目链接：http://poj.org/problem?id=2151

Description

Input

Output

Sample Input

2 2 2
0.9 0.9
1 0.9
0 0 0

Sample Output

0.972

Source

题意：

一共有 M 道题， T 个队伍，给出每个队伍做对每道题的概率，

求每个队伍最少做出来一道题和冠军至少做出 N 道题的概率！

代码如下：

#include <cstdio>
#include <cstring>
double p[1017][32];
//p[i][j]：第i支队伍做对第j道题的概率
double dp[1017][32][32];
//dp[i][j][k]:第i支队伍，前j道题中解决了k道题的概率
int main()
{
    int M, T, N;
    while(~scanf("%d%d%d",&M,&T,&N))
    {
        if(M==0 && T==0 && N==0)
        {
            break;
        }
        memset(dp,0,sizeof(dp));
        for(int i = 1; i <= T; i++)
        {
            for(int j = 1; j <= M; j++)
            {
                scanf("%lf",&p[i][j]);
            }
        }
        for(int i = 1; i <= T; i++)
        {
            dp[i][0][0] = 1;
            for(int j = 1; j <= M; j++)
            {
                dp[i][j][0] += dp[i][j-1][0]*(1-p[i][j]);
                for(int k = 1; k <= j; k++)
                {
                    dp[i][j][k] = dp[i][j-1][k-1]*p[i][j]+dp[i][j-1][k]*(1-p[i][j]);
                }
            }
        }
        double ans1 = 1, ans2 = 1;
        for(int i = 1; i <= T; i++)
        {
            ans1*=(1-dp[i][M][0]);//至少做对一道
        }
        double ans = 0;
        for(int i = 1; i <= T; i++)
        {
            double tt = 0;
            for(int k = 1; k <= N-1; k++)//只做对1~N-1道
            {
                tt+=dp[i][M][k];
            }
            ans2*=tt;
        }
        ans = ans1 - ans2;
        printf("%.3lf\n",ans);
    }
    return 0;
}

POJ 2151 Check the difficulty of problems(概率dp啊)

标签：dp   poj   

原文地址：http://blog.csdn.net/u012860063/article/details/45618709