POJ 2151 概率DP

思路：
f[i][j][k]表示i所学校 前j题 解出了k道
初始化f[i][0][0]和f[i][1][0]
f[i][j][k]=f[i][j-1][k-1]a[i][j]+f[i][j-1][k](1-a[i][j]);

这个时候f[i][n][k]表示第i所学校做出k道题的概率

我们可以维护一个
sum[i]表示第i所学校做出1<=x<=n道题的概率
kth[i]表示第i所学校做出1<=x< k道题的概率

最后把所有的sum[i]乘起来减所有kth[i]乘起来就是最后的解了

//By SiriusRen#include <cstdio>#include <cstring>#include <algorithm>using namespace std;int n,m,K;double a[1005][55],f[1005][55][55],fst[1005],kth[1005],mulfst,mulkth;//f[i][j][k]表示i所学校 前j题 解出了k道 int main(){    while(scanf("%d%d%d",&n,&m,&K)&&(n||m||K)){        memset(fst,0,sizeof(fst)),memset(kth,0,sizeof(kth)),mulfst=mulkth=1;        for(int i=1;i<=m;i++){            for(int j=1;j<=n;j++){                scanf("%lf",&a[i][j]);            }        }        for(int i=1;i<=m;i++){            f[i][0][0]=1,f[i][1][0]=1-a[i][1];            for(int j=2;j<=n;j++){                f[i][j][0]=f[i][j-1][0]*(1-a[i][j]);            }        }        for(int i=1;i<=m;i++){            for(int j=1;j<=n;j++){                for(int k=1;k<=j;k++){                    f[i][j][k]=f[i][j-1][k-1]*a[i][j]+f[i][j-1][k]*(1-a[i][j]);                }            }        }        for(int i=1;i<=m;i++){            for(int j=1;j<=n;j++){                fst[i]+=f[i][n][j];                if(j<K)kth[i]+=f[i][n][j];            }            mulfst*=fst[i];mulkth*=kth[i];        }        printf("%.3lf\n",mulfst-mulkth);    }}