CF 148 D - Bag of mice（概率DP）

🐀 🐀 🐀
dp[ i ] [ j ],i只白鼠，j只黑鼠获胜的概率
边界条件：dp[ i ] [ 0 ]=1，dp[ 0 ] [ j ]=0
题意给出的四种情况（转移方程）：
先手拿到白鼠i/(i+j)
先手拿到黑鼠，后手拿到白鼠j/(i+j) * i/(i+j-1)
先手拿到黑鼠，后手拿到黑鼠，跑掉一只黑鼠 j/(i+j) * (j-1)/(i+j-1) * (j-2)/(i+j-2) * dp[ i ] [ j - 3 ]
先手拿到黑鼠，后手拿到黑鼠，跑掉一只白鼠 j/(i+j) * (j-1)/(i+j-1) * i/(i+j-2) * dp [ i - 1 ] [ j - 2 ]

double dp[1111][1111];
int main()
{
    int w,b;cin>>w>>b;
    memset(dp,0,sizeof(dp));
    for(int i=0;i<=b;++i) dp[0][i]=0;
    for(int i=1;i<=w;++i) dp[i][0]=1;
    for(int i=1;i<=w;++i)
    {
        for(int j=1;j<=b;++j)
        {
            dp[i][j]+=(double)i/(i+j);
            if(j>=3) dp[i][j]+=(double)j/(i+j) * (double)(j-1)/(i+j-1) *(double) (j-2)/(i+j-2) * dp[i][j-3];
            if(j>=2) dp[i][j]+=(double)j/(i+j) * (double)(j-1)/(i+j-1) * (double)i/(i+j-2) * dp[i-1][j-2];
        }
    }
    printf("%.9f\n",dp[w][b]);
    //stop;
    return 0;
}