codeforces 148 D. Bag of mice(概率dp)

概率dp入门题

题意：龙和公主捉老鼠谁先捉到白老鼠谁赢 龙每捉一只老鼠会吓跑一只老鼠 公主不会吓跑老鼠 屋里共有w只白老鼠 b只黑老鼠 公主先捉 问公主赢的概率

dp[i][j] 表示公主面对i只白老鼠 j只黑老鼠赢的概率

 

题目链接：点击打开链接

#include<stdio.h>
#include<string.h>

double dp[1005][1005];

int main()
{
    int w,b,i,j;
    while(scanf("%d %d",&w,&b)!=EOF)
    {
        for(i=1;i<=w;i++)
        {
            dp[i][0]=1.0;
        }
        for(i=0;i<=b;i++)
        {
            dp[0][i]=0.0;
        }
        for(i=1;i<=w;i++)
        {
            for(j=1;j<=b;j++)
            {
                dp[i][j]=0;

                //公主抓住白鼠
                dp[i][j]+=i*1.0/(i+j);
                //公主抓住黑鼠龙抓白鼠
                dp[i][j]+=0.0;
                //公主抓黑鼠龙抓黑鼠吓走一只黑鼠
                if(j>=3)
                dp[i][j]+=(j*1.0/(i+j))*((j-1)*1.0/(i+j-1))*((j-2)*1.0/(i+j-2))*dp[i][j-3];
                //公主抓黑鼠龙抓黑鼠吓走一只白鼠
                if(j>=2)
                    dp[i][j]+=(j*1.0/(i+j))*((j-1)*1.0/(i+j-1))*(i*1.0/(i+j-2))*dp[i-1][j-2];
            }
        }
        printf("%.9f\n",dp[w][b]);
    }
    return 0;
}