题意:
在一次比赛中,有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;
}