POJ 2152 Check the difficulty of problems(三维dp)

题意:

在一次比赛中,有t只队伍参赛,总共有m道题目,现给出每支队伍解出每道题的概率,问满足以下两条件的概率为多少

条件1:每支队伍至少解出一道题目.

条件2:夺冠的队伍至少解出n道题目.

思路:

首先用p[i][j]记录第i只队伍解出第j道题的可能性.

这个题目每个队伍是一个情况,每个队伍在到第j到题为止解出k道题又是一种状态.

所以应该用一个三维dp数组来记录状态求转移方程.

有了这个数组,转移方程其实就不难想了.dp[i][j][k]=p[i][j]*dp[i][j-1][k-1]+(1-p[i][j])*dp[i][j-1][k];(黑色部分是j,很重要,因为这是在截止到第j道题目的状态)

那么dp[i][m][n]就是做出n道题的可能性了.

题目所求满足两个条件概率

设满足条件1的概率为p1

做出1到n-1的概率为p2

由于p1表示为做出1到m道题的概率.

又因为最后的答案应该为两个条件概率的交集,故所求为有队伍做出n-m道题且都做出题的概率.

所以p1-p2即有做出n-m道题且都做出题的概率,即最后的答案,

若要求p2

可开一个二维数组s[i][j],表示第i只队伍做出少于等于j题的概率.

所以s[i][j]=dp[i][m][0]+dp[i][m][1]+...+dp[i][m][j];

则p2=(s[1][n-1]-s[1][0])*....(s[t][n-1]-s[t][0]).

p1容易求出,就不讲了.

代码:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <cmath>
using namespace std;
double dp[1005][35][35];
double s[1005][35];
double p[1005][35];
int n, t, m;
int main()
{
	while (~scanf("%d%d%d", &m, &t, &n))
	{
		if (n == t && n == m && n == 0)
		{
			break;
		}
		int i, j;
		memset(dp, 0, sizeof(dp));
		memset(s, 0, sizeof(s));
		double p1 = 1, p2 = 1;
		for (i = 1; i <= t; i++)
		{
			double temp = 1;
			for (j = 1; j <= m; j++)
			{
				scanf("%lf", &p[i][j]);
				temp *= (1 - p[i][j]); //求队伍一道题都做不出的概率
			}
			p1 *= (1 - temp);
		}
		for (i = 1; i <= t; i++)
		{
			dp[i][0][0] = 1.0;
			for (j = 1; j <= m; j++)
			{
				dp[i][j][0] = 1;
				for (int k = 1; k <= j; k++)
				{
					dp[i][j][0] *= (1 - p[i][k]);
				}
			}
		}
		for (i = 1; i <= t; i++)
		{
			for (j = 1; j <= m; j++)
			{
				for (int k = 1; k <= j; k++)
				{
					dp[i][j][k] = p[i][j] * dp[i][j - 1][k - 1] + (1 - p[i][j]) * dp[i][j - 1][k];

				}
			}
		}

		for (i = 1; i <= t; i++)
		{
			for (j = 0; j <= m; j++)
				for (int k = 0; k <= j; k++)
				{
					s[i][j] += dp[i][m][k];
				}
		}
		for (i = 1; i <= t; i++)
		{
                        p2 *= (s[i][n - 1] - s[i][0]);
		}
		printf("%.3f\n", p1- p2);
	}
	return 0;
}