概率DP基础 CF148D Bag of mice

#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include 
#define CLEAR(a) memset((a),0,sizeof((a)))
using namespace std;
typedef long long LL;
const int inf=1e9;
const int maxn=1000+10;
double dp[maxn][maxn];

int main()
{
    int w,b;
	while(cin>>w>>b){
        CLEAR(dp);
		for(int i=1;i<=w;++i)dp[i][0]=1;
		for(int i=0;i<=b;++i)dp[0][i]=0;
		for(int i=1;i<=w;++i){
			for(int j=1;j<=b;++j)
			{
				dp[i][j]=i*1.0/(i+j);
				//dp[i][j]+=j*1.0/(i+j) * i*1.0/(i+j-1) * dp[i-1][j-1];
				if(j>=2)dp[i][j]+=j*1.0/(i+j) * (j-1)*1.0/(i+j-1) * i*1.0/(i+j-2) * dp[i-1][j-2];
				if(j>=3)dp[i][j]+=j*1.0/(i+j) * (j-1)*1.0/(i+j-1) * (j-2)*1.0/(i+j-2) * dp[i][j-3];
			}
		}
		printf("%.9lf\n",dp[w][b]);
	}
	return 0;
}

你可能感兴趣的:(ACM)