atcoder D - increment of coins（概率DP）

题目链接：D - increment of coins

题意： 当前袋子里有金币A枚，银币B枚，铜币C枚，如果袋子里某种硬币超过100个，则结束。每次操作从袋子里等概率的拿出一枚硬币，然后将两个同类硬币放回袋子。求操作次数的期望。(A,B,C<=99)

分析：这就是一道比较经典的概率DP题目，设f[i][j][k]表示当前有i枚A硬币，j枚B硬币，k枚C硬币时的期望次数。

我们下面来看一下如何进行状态转移：
假设当前有i枚A硬币，j枚B硬币，k枚C硬币，那么下一次取出A硬币的概率就是A/(A+B+C),同理取出B硬币和C硬币的概率分别是B/(A+B+C)和C/(A+B+C),那么我们当前状态的期望次数就是我们下一次枚举状态的期望次数*其转移概率再加上当前状态转移所需要的一次次数。也就是说

f[A][B][C]=1+A/(A+B+C)*f[A+1][B][C]+B/(A+B+C)*f[A][B+1][C]+C/(A+B+C)*f[A][B][C+1]

由于我们当前转态需要用到的状态还没有更新，所以我们可以用记忆化搜索来进行更新，边界条件就是A，B，C其中一个值大于等于100，细节见代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<vector>
#include<queue>
using namespace std;
const int N=1e2+10;
double f[N][N][N];
//f[i][j][k]当前有i枚A硬币，j枚B硬币，k枚C硬币时的期望次数 
double dfs(int x,int y,int z)
{
	if(x>=100||y>=100||z>=100) return 0;
	if(f[x][y][z]>0) return  f[x][y][z];
	f[x][y][z]=1.0;
	f[x][y][z]+=1.0*x/(x+y+z)*dfs(x+1,y,z);
	f[x][y][z]+=1.0*y/(x+y+z)*dfs(x,y+1,z);
	f[x][y][z]+=1.0*z/(x+y+z)*dfs(x,y,z+1);
	return f[x][y][z];
}
int main()
{
	int n,m,l;
	cin>>n>>m>>l;
	printf("%.10lf",dfs(n,m,l));
	return 0;
}