期望题集--CF148D Bag of mice

最新推荐文章于 2022-01-20 08:55:05 发布

caoyang1123

最新推荐文章于 2022-01-20 08:55:05 发布

阅读量186

点赞数

分类专栏：刷题集

本文链接：https://blog.csdn.net/caoyang1123/article/details/83385802

版权

刷题集专栏收录该内容

110 篇文章 4 订阅

订阅专栏

题意：袋子里有ww 只白鼠和bb 只黑鼠，A和B轮流从袋子里抓，谁先抓到白色谁就赢。A每次随机抓一只，B每次随机抓完一只之后会有另一只随机老鼠跑出来。如果两个人都没有抓到白色则B赢。A先抓，问A赢的概率。

直接dfs记忆化搜索，状态记录剩下两种球个数分别为多少且轮到谁时A赢的期望

代码：

#include<bits/stdc++.h>
#define db double
using namespace std;
const int N=1010;
db dp[N][N][2];
bool vis[N][N][2];
db dfs(int a,int b,bool op)
{
    if(a<0||b<0)return 0;
    if(vis[a][b][op])return dp[a][b][op];
    vis[a][b][op]=1;
    if(!b)return dp[a][b][op]=0;
    if(op==0)
    {
        if(!a)return dp[a][b][op]=1;
        return dp[a][b][op]=1.0*b/(a+b)+1.0*a/(a+b)*dfs(a-1,b,op^1);
    }
    else
    {
        if(!a)return dp[a][b][op]=0;
        db tp=1.0*a/(a+b);
        return dp[a][b][op]=tp*(1.0*(a-1)/(a+b-1)*dfs(a-2,b,op^1)+1.0*b/(a+b-1)*dfs(a-1,b-1,op^1));
    }
}
int main()
{
    int a,b;
    cin>>b>>a;
    printf("%.10lf",dfs(a,b,0));
}