占坑[概率dp] poj 2096 Collecting Bugs

题目分析

    dp[i][j]代表已达到前i和前j，以此为起始到达n、s的期望。
    此时，有$p_{i1}=\frac{i}{n}$的概率再次选择前i个，$q_{j1}=\frac{j}{s}$的概率再次选择前j个，$p_{i2}=\frac{n-i}{n}$的概率选择后面n-i个中的一个，$q_{j2}=\frac{s-j}{s}$的概率选择后面s-j个中的一个。则$dp_{ij}=p_{i1}q_{j1}dp_{ij}+p_{i1}q_{j2}dp_{i(j+1)}+p_{i2}q_{j1}dp_{(i+1)j}+p_{i2}q_{j2}dp_{(i+1)(j+1)+1}$
    “+1”要放在外面，而非等式右边每个的dp上。(占坑日后解释)

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<map>
#include<algorithm>
using namespace std;
const double eps=1e-9;
const int maxn = 1003;
int n,s;
double ji;
double dp[maxn][maxn];
int main()
{
    while(cin>>n>>s)
    {
        ji=n*s;
        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 x,y;
            x=1-i*j/ji;
            y=1;
            if(i+1<=n) y+=(n-i)*j/ji*dp[i+1][j];
            if(j+1<=s) y+=i*(s-j)/ji*dp[i][j+1];
            if(i+1<=n && j+1<=s)  y+=(n-i)*(s-j)/ji*dp[i+1][j+1];
            dp[i][j]=y/x;
        }
        printf("%.4f\n",dp[0][0]);
    }
    return 0;
}