poj2151(概率dp)

链接：点击打开链接

题意：有M道题，T个队，给出每个队做出每道题的概率，求每队至少做出一题并且冠军至少做出N道题的概率

代码：

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
double f[1005][35],dp[1005][35][35];
int main(){                                     //dp[i][j][k]表示第i个队前j道题做k道的概率
    int M,T,N,i,j,k;
    double ans,tmp,sum;
    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",&f[i][j]);
        memset(dp,0,sizeof(dp));
        for(i=1;i<=T;i++)
        dp[i][0][0]=1;
        for(i=1;i<=T;i++){
            for(j=1;j<=M;j++){
                for(k=0;k<=j;k++){
                    if(k>=1)                    //做出第i道题
                    dp[i][j][k]+=dp[i][j-1][k-1]*f[i][j];
                    if(j-1>=k)                  //没做出第i道题
                    dp[i][j][k]+=dp[i][j-1][k]*(1-f[i][j]);
                }
            }
        }
        ans=tmp=1;
        for(i=1;i<=T;i++)                       //每个人至少做一题的概率
        ans*=(1-dp[i][M][0]);
        for(i=1;i<=T;i++){
            sum=0;
            for(j=1;j<N;j++)                    //
            sum+=dp[i][M][j];
            tmp*=sum;
        }
        ans-=tmp;                               //减去没人做至少N题的概率
        printf("%.3lf\n",ans);
    }
    return 0;
}