Collecting Bugs（POJ2096）概率dp

题意：有s个子系统和n个bug，程序员每天会发现一个随机的bug（bug可重复），求在s个子系统中发现n个bug所需要时间的数学期望

分析：

我们用dp[i][j]来表示已经在第j个子系统中发现i类bug所需要天数的数学期望，那么可以知道dp[n][s]=0（目标已达成）,而dp[0][0]则为所求；

dp[i][j]状态可以转化成以下四种：

dp[i+1][j+1]发现一个bug属于新的一种bug和新的一个子系统 
dp[i+1][j] 发现一个bug属于新的一种bug，但属于已经找到的j种子系统 
dp[i][j+1] 发现一个bug属于已经找到的i种bug，但属于新的子系统 
dp[i][j] 发现一个bug属于已经找到的i种bug和j个子系统中 

以上四种的概率分别为： 

p1 = (n-i)*(s-j) / (n*s)         //(1-i/n)*(1-j/s)
p2 = (n-i)*j / (n*s)              //(i/n)*(1-j/s)
p3 = i*(s-j) / (n*s)              //(1-i/n)*(j/s)
p1 = i*j / (n*s)                   //(i/n)*(j/s)

于是，我们的状态转移方程就出炉了，同样根据期望的定义，

dp[i][j] = p1*dp[i+1][j+1] + p2*dp[i+1][j] + p3*dp[i][j+1] + p4*dp[i][j] + 1

移项合并一下，

dp[i][j] = (p1*dp[i+1][j+1] + p2*dp[i+1][j] + p3*dp[i][j+1] + 1) / (1-p4).

复杂度O（n*s）

AC代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<vector>
#include<utility>
#include<string>
#include<queue>
#include<map>
#include<set>
#include<iterator>
#include<iomanip>
#include<stack>
#include<cstdio>
#define endl "\n"
using namespace std;
int n,s;
const int maxn=1010;
double dp[maxn][maxn];
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);cout.tie(0);
    double p1,p2,p3,p4;
    while(cin>>n>>s)
    {
    	memset(dp,0,sizeof dp);
    	for(int i=n;i>=0;i--)
    	{
    		for(int j=s;j>=0;j--)
    		{
    			if(i==n && j==s)
                    continue;
    			p1=(1.0*(n-i)*(s-j))/(1.0*n*s);
    			p2=(1.0*(n-i)*j)/(1.0*n*s);
    			p3=(1.0*i*(s-j))/(1.0*n*s);
    			p4=(1.0*i*j)/(1.0*n*s);
    			dp[i][j] = (p1*dp[i+1][j+1] + p2*dp[i+1][j]+p3*dp[i][j+1]+1.0)/ (1.0-p4);
			}
		}
		cout<<fixed<<setprecision(4)<<dp[0][0]<<endl;
	}
    return 0;
}