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;

}