POJ 2096 概率DP入门

【题目链接】
http://acm.hust.edu.cn/vjudge/problem/viewProblem.action?id=21521

【解题报告】
一共有n种bug，s个系统，每天发现一个bug（一共有无限多个bug，bug的种类和属于哪个系统是独立随机事件），问至少找齐
n种bug，每个系统至少找到一个bug的期望天数。

我们使用递推的方法来解决这道题目。
设dp[i][j]表示已找到n种bug，并且已有s个系统里有bug的期望。
假如再过一天我们找到一个bug，那么这是一个离散型随机变量。
下一个状态只可能是dp[i][j],dp[i+1][j],dp[i+1][j+1],dp[i][j+1]这四种状态中的一种。
于是我们知道dp[i][j]= P1*dp[i][j] + P2*dp[i+1][j] + P3*dp[i+1][j+1] + P4*[i][j+1]+1.
之所以+1是因为dp[i][j]这个状态比后面的状态多一天。

【参考代码】

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
#include<cmath>
#include<iomanip>
using namespace std;

const int maxn=1010;
double dp[maxn][maxn];
int n,s;

//找到n种类型的bug，并且s个类型中至少有一个
int main()
{
      while( cin>>n>>s )
      {
            memset( dp,0,sizeof dp );
            dp[n][s]=0;
            for( int i=n;i>=0;i-- )
            {
                  for( int j=s;j>=0;j-- )
                  {
                        if( i==n && j==s )continue;
                        double p1=i*1.0/n,p2=j*1.0/s;
                        dp[i][j]+=dp[i][j+1]*p1*(1-p2) + dp[i+1][j]*(1-p1)*p2 + dp[i+1][j+1]*(1-p1)*(1-p2)+1;
                        dp[i][j]/=(1-p1*p2);
                  }
            }
            cout<<fixed<<setprecision(6)<<dp[0][0]<<endl;
      }

      return 0;
}