POJ 2151 Check the difficulty of problems(확률 DP)

1620 단어
제목 링크: Check the difficulty of problems
문제 풀이 사고방식: 분류에서hash로 나뉘었지만 아무리 생각해도 납득이 안 간다. 마지막으로 디스커스에 있는 이것은 확률+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
", p1 - p2); } return 0; }

좋은 웹페이지 즐겨찾기