POJ 2151 Check the difficulty of problems(概率DP)

题目链接:Check the difficulty of problems

解题思路:分类里面分到了hash,但是怎么都想不通,最后看了discuss里面的这是概率+DP,概率就是拿他每一队做出题的概率(1 - 每一题都做不出)相乘再  减去    每队做出的题都没有达到期望n的概率的积。但是算题目的概率的话组合方式有很多种,包含了很多不必要的乘法步骤,暴力求解的话肯定会超时。看了一个人的博客找到了状态转移方程DP[i][j][k] = DP[i][j - 1][k - 1] * P[J] + DP[i][j - 1][k] * (1 - P[J]) (i表示队伍数, j表示当前题目数, k表示解题数)

#include<stdio.h>
#include<string.h>

double dp[1010][32][32];
double ans[1010][2];
double pro[32]; 

int main(){
	int i, j, k, n, m, t;
	//freopen("in.txt", "r", stdin); 
	while(scanf("%d%d%d", &m, &t, &n) && (m || t || n)){
		memset(dp, 0, sizeof(dp));
		memset(ans, 0, sizeof(ans)); 
		for(i = 1; i <= t; i++){
			dp[i][0][0] = 1;
			for(j = 1; j <= m; j++){
				scanf("%lf", &pro[j]);
			}
			for(j = 1; j <= m; j++){
				for(k = 0; k <= j; k++){
					if(k - 1 >= 0)
						dp[i][j][k] = (dp[i][j - 1][k] * (1 - pro[j])) + (dp[i][j - 1][k - 1] * pro[j]);
					else
						dp[i][j][k] = (dp[i][j - 1][k] * (1 - pro[j]));
				}
			}
			ans[i][0] = 1 - dp[i][m][0];
			for(j = 1; j < n; j++){
				ans[i][1] += dp[i][m][j];
			}
		}
		double p1 = ans[1][0], p2 = ans[1][1];
		for(i = 2; i <= t; i++){
			p1 *= ans[i][0];
			p2 *= ans[i][1];
		}
		printf("%.3lf\n", p1 - p2);
	}
	return 0;
}


 

你可能感兴趣的:(POJ 2151 Check the difficulty of problems(概率DP))