BZOJ 1419: red is good

1419: Red is good

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

Description

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

Input

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

Output

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

Sample Input

5 1

Sample Output

4.166666

HINT

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

先来构思dp式，dp[i][j]表示剩余i红j黑

逆序求期望 so 在代码里找di

再搞一个滚动数组压缩空间

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<iomanip>
using namespace std;
int n,m;
double dp[5002];
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
    {
        dp[0]=i;
        for(int j=1;j<=m;j++)
        {
            dp[j]=max(0.0,1.0*(j*(dp[j-1]-1)+i*(dp[j]+1))/(i+j));
        }
    }
    printf("%f",dp[m]-5e-7);
}