poj 2096Collecting Bugs

题目链接

poj 2096Collecting Bugs

题解

dp[i][j]表示已经找到i种bug，并存在于j个子系统中，要达到目标状态的天数的期望。
显然，dp[n][s]=0，因为已经达到目标了。而dp[0][0]就是我们要求的答案。
dp[i][j]状态可以转化成以下四种：
dp[i][j] 发现一个bug属于已经找到的i种bug和j个子系统中-->p1 = ij / (ns)
dp[i+1][j] 发现一个bug属于新的一种bug，但属于已经找到的j种子系统-->p2 = (n-i)j / (ns)
dp[i][j+1] 发现一个bug属于已经找到的i种bug，但属于新的子系统-->p3 = i(s-j) / (ns)
dp[i+1][j+1]发现一个bug属于新的一种bug和新的一个子系统-->p4 = (n-i)(s-j) / (ns)
dp[i][j] = \(\sum 由XX转移来的期望\)
发现dp[i][j]不能用dp[i][j] 更新，移项有去除dp[i][j]列转移方程

代码

#include<cstdio>
#include<algorithm> 
inline int read() { 
	int x = 0; 
	char c = getchar(); 
	while(c < '0' || c > '9')c = getchar(); 
	while(c <= '9' && c >= '0')x = x * 10 + c - '0',c = getchar(); 
	return x; 
} 
const int maxn = 1007; 
int n,s;  
double dp[maxn][maxn]; 
int main() { 
	n = read();s = read(); 	
	for(int i = n;i >= 0;-- i) {
		for(int j = s;j >= 0;-- j) {
	 		if(i == n && s == j) continue; 
			double p1 = 1.0 * i * j / (1.0 * n * s); 
			double p2 = 1.0 * (n - i) * j / (1.0 * n * s); 
			double p3 = 1.0 * i * (s - j) / (1.0 * n * s); 
			double p4 = 1.0 * (n - i) * (s - j) / (1.0 * n * s); 
			dp[i][j] = double (dp[i + 1][j] * p2 + dp[i][j + 1] * p3 + dp[i + 1][j + 1] * p4 + 1) / (1 - p1); 
		}
	}	
	printf("%.4lf\n",dp[0][0]); 
	return 0; 
}