CodeForces 148D : Bag of mice概率DP

匿枫

于 2021-08-05 21:10:16 发布

阅读量132

点赞数

分类专栏：大学ACM

本文链接：https://blog.csdn.net/tlyzxc/article/details/119427200

版权

概率动态规划游戏策略随机过程 C++

关键词由CSDN通过智能技术生成

大学ACM 专栏收录该内容

162 篇文章 5 订阅

订阅专栏

传送门

题意

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

分析

这道题我们可以去顺推，一开始有 $n$ 只小白鼠和 $m$ 只小黑鼠，那么我们就有两种决策，选小白鼠的话，结束游戏，最后答案+ $n / (m + n)$ ，选小黑鼠，又产生两种决策，最后结束的条件是， $n = 0$ ，B赢， $m = 0$ ，A赢，跑一下记忆化搜索即可

代码

#pragma GCC optimize(3)
#include <bits/stdc++.h>
#define debug(x) cout<<#x<<":"<<x<<endl;
#define dl(x) printf("%lld\n",x);
#define di(x) printf("%d\n",x);
#define _CRT_SECURE_NO_WARNINGS
#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;
typedef vector<int> VI;
const int INF = 0x3f3f3f3f;
const int N = 1010;
const ll mod = 1000000007;
const double eps = 1e-9;
const double PI = acos(-1);
template<typename T>inline void read(T &a) {
	char c = getchar(); T x = 0, f = 1; while (!isdigit(c)) {if (c == '-')f = -1; c = getchar();}
	while (isdigit(c)) {x = (x << 1) + (x << 3) + c - '0'; c = getchar();} a = f * x;
}
int gcd(int a, int b) {return (b > 0) ? gcd(b, a % b) : a;}

double f[N][N];

double res(int x,int y){  //白兔子，黑兔子
	if(!x) return 0;
	if(!y) return 1.0;
	if(f[x][y]) return f[x][y];
	double ans = 1.0 * x / (x + y);
	if(y == 2) ans += 1.0 * y /(x + y) * (y - 1) / (x + y - 1);
	else if(y >= 3)ans += 1.0 * y / (x + y) * (y - 1) / (x + y - 1) * (1.0 * (x) / (x + y - 2) * res(x - 1,y - 2) + 1.0 *(y - 2) / (x + y - 2) * res(x,y - 3));
	f[x][y] = ans;
	return ans;
}

int main() {
	int n,m;
	read(n),read(m);
	printf("%.9lf\n",res(n,m));
	return 0;
}