poj-2151

Check the difficulty of problems

    题意：第一行输入M,T,N三个参数。M表示一共有M个问题，T表示一共有T个参赛队伍，N表示冠军至少要做对几道题。接下来T行，表示第1队到第T队做对每一道题目的概率。要求至少每队做出一道题并且至少有一队做出N道题的概率。

    用dp[i][j][k]表示第i队，前j道题做对k题的概率，那么dp[i][j+1][k]这个状态可以由两种状态转换而来。由dp[i][j-1][k]状态再做错第j题转换来，也就是  dp[i][j][k] = dp[i][j-1][k]*(1-p[j])，p[j]表示做对第j题的概率。还可以从dp[i][j-1][k-1]状态再做对第j题转换来，也就是dp[i][j][k]+=dp[i][j-1][k-1]*p[j]。因此得到可动归得到每个队伍，M道题中，做对1~M个题目的概率。

    要求至少每队做出一道题并且至少有一队做出N道题的概率，可以转化为所有队都做对一道题的概率，减去所有队做对1~N-1道题的概率。

    AC的代码如下：

#include<iostream>
#include<stdio.h>
#include<cstring>
using namespace std;
float dp[1005][32][32]; 
int main()
{
	int m,t,n;
	float p[33];
	cin >> m >> t >> n;
	memset(dp,0.0,sizeof(dp));
	for(int i=0;i<=1000;i++)
	{
		dp[i][0][0]=1;
	}
	while(m!=0&&t!=0&&n!=0)
	{
		for(int i=1;i<=t;i++)
		{
			for(int l=1;l<=m;l++)
			{
				cin >> p[l];
			}
			for(int j=1;j<=m;j++)
			{
				for(int k=0;k<=j;k++)
				{
					dp[i][j][k]=dp[i][j-1][k]*(1-p[j]);
					if(k>0)
					{
						dp[i][j][k]+=dp[i][j-1][k-1]*p[j];//第i队前j道题，做对k道 
					}
				} 
			}
		}
		double a=1.0,b=1.0;
		for(int i=1;i<=t;i++)
		{
			a = a*(1.0-dp[i][m][0]);
			double temp=0;
			for(int k=1;k<n;k++)
			{
				temp += dp[i][m][k];
			}
			b = b*temp;
		}
		printf("%.3f\n",a-b);
		cin >> m >> t >> n;
	}
	return 0;
}