POJ 2151 Check the difficulty of problems

标签：思维   概率dp   poj   

题目链接：POJ 2151 Check the difficulty of problems

题面：

Description

Input

Output

Sample Input

2 2 2
0.9 0.9
1 0.9
0 0 0

Sample Output

0.972

Source

第一次写概率DP，感觉和DP有些不一样，借鉴了这篇博客：金海峰

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
double correspond[1005][31];
double dp[1005][31][31];
int main()
{
	int n,m,t;
	double ans=1,tmp,sum;
	while(scanf("%d%d%d",&m,&t,&n)&&(m||t||n))
	{
		ans=tmp=1;
		for(int i=0;i<t;i++)
		{
			for(int j=1;j<=m;j++)
			  scanf("%lf",&correspond[i][j]); 
		}
		memset(dp,0,sizeof(dp)); 
		//dp[i][j][k] 第i只队伍，做前j题，做对k题的概率 
		for(int i=0;i<t;i++)
	     {
          dp[i][0][0]=1;
          for (int j=1;j<=m;j++)
          {
       	     //还是做对0道，那么就是当前这一道没做对 
             dp[i][j][0] = dp[i][j -1][0] * (1- correspond[i][j]);
             for (int k =1; k <= j; k++)
               //做对k道，要么是之前就做对了k道，这次没对，要么就是之前对了k-1道，这次对了 
               dp[i][j][k] = dp[i][j -1][k -1] * correspond[i][j]+ dp[i][j -1][k] * (1- correspond[i][j]);
          }
		 }
	    //排除任何一个人一道都没做对的情况 
        for(int i=0;i<t;i++)
          ans *=(1- dp[i][m][0]);
        //排除所有人都只做对1~n-1的情况 
       for(int i=0;i<t;i++)
        {
          sum=0;
          for(int j=1;j<n;j++)
          sum+=dp[i][m][j];
          tmp*=sum;
        }
       ans-=tmp;
       printf("%.3f\n", ans);
	}
	return 0;
} 

POJ 2151 Check the difficulty of problems

标签：思维   概率dp   poj   

原文地址：http://blog.csdn.net/david_jett/article/details/45588013