POJ 2151.Check the difficulty of problems

题目:http://poj.org/problem?id=2151

AC代码(C++):

#include <iostream>
#include <algorithm>
#include <stdio.h>
#include <vector>
#include <queue>
#include <math.h>
#include <string>
#include <string.h>
#include <bitset>

#define INF 0xfffffff
#define MAXN 100005
#define prime 99991

using namespace std;

int m,t,n;
double x[35][1005];
double p[35];

int main(){
	while(cin>>m>>t>>n){
		if(m==0)break;
		double p1 = 1, p2 = 1;
		for(int i = 0; i < t; i++){
			double p1i = 1;
			for(int j = 0; j < m; j++){
				cin>>p[j];
				p1i*=(1-p[j]);
			}
			p1*=(1-p1i);
			
			memset(x,0,sizeof(x));
			x[0][0] = 1;
			double p2i = 0;
			for(int j = 1; j <= m; j++){
				x[j][0] = x[j-1][0] * (1-p[j-1]);
				for(int k = 1; k <= n-1; k++)x[j][k] = x[j-1][k-1]*p[j-1] + x[j-1][k]*(1-p[j-1]);
			}
			for(int j = 1; j <= n-1; j++)p2i += x[m][j];
			p2*=p2i;
		}
		printf("%.3lf\n",p1-p2);
	}
}

总结: 动态规划问题. 问题求所有队伍做出至少1题且存在队伍做出至少n道题, 即求所有队伍做出至少一道题的概率p1减去所有队伍做出的题的数目都在[1,n-1]内的概率p2(也可以不包括1来直接求出答案). 用动态规划来描述就是, 某队伍在前j道题做出k道的概率为x[j][k] = x[j-1][k-1]*p[j] + x[j-1][k]*(1 - p[j]). 初始化时x数组第0行除了第0列外全赋值为0, x[0][0]为1. 逐行计算, 每行第0列可直接求出. 直到求出x[m][n-1].