期望题集--CF148D Bag of mice

xiaoxiao2025-10-16  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)); }

 

转载请注明原文地址: https://www.6miu.com/read-5038039.html

最新回复(0)