题目很简单:公主和龙在玩游戏,一个袋子里有 w 只白老鼠和 b 只黑老鼠,公主先从袋子中取一只老鼠,然后龙再取,然后会有一只老鼠等概率的跳出来,然后再由公主取,谁先取到白老鼠谁就获胜,取老鼠也是等概率的,如果取到最后都没有人取到白老鼠,则龙赢,求公主赢的概率。PS:公主先取。
概率正在求,期望倒着求。
那么就正在推下:
设dp[i][j],是公主在面对这个袋子里有 i 只白老鼠,j 只黑老鼠的情况下获胜的概率,那么dp[w][b] 就是答案了;
假设在现在公主直接取了一只白老鼠,公主就获胜了,那么 dp[i][j] =i/(i+j).
如果现在公主取的是黑老鼠,那么可以有前面已经解决的子问题求得,可以分为两种情况:
1.公主取黑老鼠,龙取黑老鼠,跳出的也是黑老鼠,问题就转移到了dp[i][j-3],那么转移的概率就成了:
dp[i][j]+=j/(i+j)(j-1)/(i+j-1)(j-2)/(i+j-2)*dp[i][j-3]; j>=3;
2.公主取黑老鼠,龙取黑老鼠,跳出的是白老鼠,那么问题就转移到了dp[i-1][j-2],那么
dp[i][j]+=j/(i+j)*(j-1)/(i+j-1)*i/(i+j-2)*dp[i-1][j-2] j>=2;
转移就已经想完了,再来考虑边界:
1.当袋子里面只有白老鼠,那么无论公主怎么取,获胜的概率就是 1。
2.当袋子里面只有黑老鼠,或者没有老鼠,公主获胜的概率都为 0。
#include<algorithm>
#include<string>
#include<string.h>
#include<queue>
#include<vector>
#include<stack>
#include<math.h>
#include<map>
#include<stdio.h>
#include<iostream>
using namespace std;
#define maxn 1050
double dp[maxn][maxn];
int b,w;
int main()
{
scanf("%d%d",&w,&b);
for(int i=1;i<=w;i++)
dp[i][0]=1.0;
for(int i=1;i<=w;i++)
{
for(int j=1;j<=b;j++)
{
dp[i][j]=(double)i/(double)(i+j);
if(j>=3)
dp[i][j]+=(double)j*(double)(j-1)*(double)(j-2)*dp[i][j-3]/((double)(i+j)*(double)(i+j-1)*(double)(i+j-2));
if(i>=1&&j>=2)
dp[i][j]+=(double)j*(double)(j-1)*(double)i*dp[i-1][j-2]/((double)(i+j)*(double)(i+j-1)*(double)(i+j-2));
}
}
printf("%.9f\n",dp[w][b]);
return 0;
}