POJ2096 Collecting Bugs

题目链接

题意

有n个bug和s个系统。每天会等概率的在某个系统中找一个某一个bug。问找到全部n个bug并且在每个系统中都找到bug的期望天数。

思路

f[i][j]表示在j个系统中找到了i个bug的期望天数。

某天有四种可能：
1.在新的系统中找到新的bug，概率为\(\frac{(n-i)\times(s-j)}{n\times s}\)
2.在旧的系统中找到新的bug，概率为\(\frac{(n-i)\times j}{n\times s}\)
3.在新的系统中找到旧的bug，概率为\(\frac{i \times (s-j)}{n\times s}\)
4.在旧的系统中找到旧的bug，概率为\(\frac{i \times j}{n\times s}\)

即$$f[i][j]= \
f[i+1][j+1] \times \frac{(n-i)\times(s-j)}{n\times s} + \
f[i + 1][j] \times \frac{(n-i)\times j}{n\times s} + \
f[i][j + 1] \times \frac{i \times (s-j)}{n\times s} + \
f[i][j] \times \frac{i \times j}{n\times s} + 1

\[ 左右都有$f[i][j]$，移项并同乘$n\times s$得 $$f[i][j] \times (n\times s -i \times j)= \\ f[i+1][j+1] \times (n-i)\times(s-j) + \\ f[i + 1][j] \times (n-i)\times j + \\ f[i][j + 1] \times i \times (s-j) + \\ n \times s \]

\[f[i][j] =\frac{f[i+1][j+1] \times (n-i)\times(s-j) +f[i + 1][j] \times (n-i)\times j + f[i][j + 1] \times i \times (s-j) + n \times s}{n \times s} \]

转移即可。lj POj的评测姬，在\(G++\)下输出\(lf\)会\(wa\)

代码

/*
* @Author: wxyww
* @Date:   2019-05-14 15:24:19
* @Last Modified time: 2019-05-14 15:59:55
*/
#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
#include<ctime>
using namespace std;
typedef long long ll;
const int N = 1010;
ll read() {
	ll x=0,f=1;char c=getchar();
	while(c<'0'||c>'9') {
		if(c=='-') f=-1;
		c=getchar();
	}
	while(c>='0'&&c<='9') {
		x=x*10+c-'0';
		c=getchar();
	}
	return x*f;
}
double f[N][N];
int main() {
	int n,s;
	while(~scanf("%d%d",&n,&s)) {
		for(int i = n;i >= 0;--i) {
			for(int j = s;j >= 0;--j) {
				if(i == n && s == j) continue;
				f[i][j] = 
					(f[i + 1][j + 1]* (n - i) * (s - j)  + //在新系统里发现新漏洞
					f[i + 1][j] * (n - i) * j  + //在旧系统里发现新漏洞
					f[i][j + 1]* i * (s - j)  + //在新系统里发现旧漏洞
					+ n * s) //在旧系统发现旧漏洞
					/ (n * s - i * j );
			}
		}
		printf("%.4lf\n",f[0][0]);
	}
	return 0;
}