【BZOJ-1419】Red is good 概率期望DP

1419: Red is good

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

Description

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

Input

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

Output

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

Sample Input

5 1

Sample Output

4.166666

HINT

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

Source

Solution

概率期望DP

$f[i][j]$表示选$i$张红和$j$张黑的答案

转移就是期望*概率

$f[i][j]=max(0,\frac{i}{i+j}*(f[i-1][j]+1)+\frac{j}{i+j}*(f[i][j-1]-1))$

卡内存需要滚动数组

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
int R,B,now;
double f[][];
int main()
{
    scanf("%d%d",&R,&B);
    for (int i=; i<=R; i++,now^=,f[now][]=i)
        for (int j=; j<=B; j++)
            f[now][j]=max(*1.0,1.0*i/(i+j)*(f[now^][j]+)+1.0*j/(i+j)*(f[now][j-]-));
    long long ans=f[R&][B]*;
    printf("%lf",(double)ans/);
    return ;
}