CodeForces 148D Bag of mice(概率DP)

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<string.h>
#include<cstring>
#include<string>
#include<map>
#include<set>
#include<vector>
using namespace std;
typedef long long ll;
int ans[120];
#define sp  system("pause")
#define mt(dp) memset(dp,0,sizeof dp)

double dp[1200][1200];


int main()
{
	int w, b;
	cin >> w >> b;
	mt(dp);
	dp[w][b] = 1;
	double ans = 0;
	int sum = w + b;
	for (int i = 0; i <= sum; i++)
	{
		for (int j = 0; j <= i; j++)
		{
			if (w - j >= 0 && b - (i - j) >= 0 && sum - i != 0)
				ans += (dp[w - j][b - (i - j)] * 1.0*(w - j) / (sum - i));
			/*
			if(w-j-1>=0)
			{
			dp[w-j-1][b-(i-j)]+=(dp[w-j][b-(i-j)]*(w-j)/(sum-i));
			}
			*/
			if (b - (i - j) - 1 >= 0 && w - j >= 0)
			{
				dp[w - j][b - (i - j) - 1] += (dp[w - j][b - (i - j)] * (b - (i - j)) / (sum - i));
			}
		}
		i++;
		if (i>sum)break;
		for (int j = 0; j <= i; j++)
		{
			if (b - (i - j) - 1 >= 0 && (w - j) >= 0 && sum - i != 0)
			{
				dp[w - j][b - (i - j) - 1] += (dp[w - j][b - (i - j)] * (b - (i - j)) / (sum - i));
			}
		}
		i++;
		if (i>sum)break;
		for (int j = 0; j <= i; j++)
		{
			if (sum - i != 0)
			{
				if (w - j - 1 >= 0 && (b - (i - j)) >= 0)
				{
					dp[w - j - 1][b - (i - j)] += (dp[w - j][b - (i - j)] * (w - j) / (sum - i));
				}
				if (b - (i - j) - 1 >= 0 && (w - j) >= 0)
				{
					dp[w - j][b - (i - j) - 1] += (dp[w - j][b - (i - j)] * (b - (i - j)) / (sum - i));
				}
			}

		}
	}
	if (!w&&!b)
		printf("%.9lf\n", 0);
	else
		printf("%.10f\n", ans);
	//sp;
	return 0;

}