POJ 2151--Check the difficulty of problem

• 概率DP，proba[i][j]:前i个题目解出j道的概率。
• 根据此列出状态转移方程：proba[i][j] = proba[i-1][j-1]*p[i]+proba[i-1][j](1-p[i]).其中p[i]代表解出第i个题目的概率。
• 题目求：每个队解出至少一个题目，其中冠军(解出最多题目的队伍)解出至少N个题目。将以上事件分为A,B。则P(AB) = P(A)-P(A~B)，感性地理解，即每个队解出至少一个题目的概率，减去每个队都只解出1到N-1个题目的概率，这就能保证至少一个队伍解出多余N个题目。

#include<cstdio>
#include<cstring>
using namespace std;
#define maxT 1001
#define maxM 31

float probability[maxT][maxM];
int teamNum,problemNum,championNum;
float proba[maxM][maxM];

int calc_probability()
{
    int i,j,k;
    float ansProba,tmpProba,proba1toN;
    ansProba = tmpProba = 1;
    proba[0][0] = 1;
    for(i = 0;i < teamNum;i++)
    {
    for(j = 1;j <= problemNum;j++)
    {
    proba[j][0] = proba[j-1][0]*(1-probability[i][j-1]);     //前j道题做出0道概率
    for(k = 1;k <= j&&k <= championNum-1;k++)
        proba[j][k] = proba[j-1][k-1]*probability[i][j-1]+proba[j-1][k]*(1-probability[i][j-1]);      //第i支队伍在前j道题做出k道的概率
    }
    ansProba *= (1-proba[problemNum][0]);          //每个队解出至少一个题目概率
    proba1toN = 0;
    for(k = 1;k <= championNum-1;k++)
    proba1toN += proba[problemNum][k];
    tmpProba *= proba1toN;                  //每个队都解出1到N-1个题目概率
    }
    ansProba -= tmpProba;
    printf("%.3f\n",ansProba);
    return 0;
}

int main()
{
    int i,j;
    while(scanf("%d%d%d",&problemNum,&teamNum,&championNum))
    {
        if(!problemNum&&!teamNum&&!championNum)
            return 0;
        for(i = 0;i < teamNum;i++)
        {
            for(j = 0;j < problemNum;j++)
            scanf("%f",probability[i]+j);
        }
        calc_probability();
    }
    return 0;
}