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;
}