[bzoj1419]Red is good 期望DP

1419: Red is good

Time Limit: 10 Sec   Memory Limit: 64 MB
[ Submit][ Status][ Discuss]

Description

桌面上有R张红牌和B张黑牌，随机打乱顺序后放在桌面上，开始一张一张地翻牌，翻到红牌得到1美元，黑牌则付出1美元。可以随时停止翻牌，在最优策略下平均能得到多少钱。

Input

一行输入两个数R,B,其值在0到5000之间

Output

在最优策略下平均能得到多少钱。

Sample Input

5 1

Sample Output

4.166666

HINT

输出答案时,小数点后第六位后的全部去掉,不要四舍五入.

Source

f[i][j]表示还剩i红，j黑时的期望

保留后六位就乱搞一下

#include<iostream>
#include<cstdio>
using namespace std;
const int N = 5000 + 5;
double f[2][N]; int R, B;
int main(){
	scanf("%d%d", &R, &B);
	for( int i = 0; i <= R; i++ ){
		f[i&1][0] = i;
		for( int j = 1; j <= B; j++ ){
			double p = (double)i/(i+j);
			f[i&1][j] = max(0.0,(f[i&1][j-1]-1)*(1-p)+(f[(i-1)&1][j]+1)*p);
		}
	}
	f[R&1][B] *= 1000000.0;//乱写的 
	int yu = (int)f[R&1][B];
	f[R&1][B] = (double)yu<=f[R&1][B] ? (double)yu : (double)yu-1.0;
	f[R&1][B] /= 1000000.0;
	printf("%lf",f[R&1][B]);
	return 0;
}