1419: Red is good
Time Limit: 10 Sec Memory Limit: 64 MBSubmit: 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);
}