【bzoj1419】Red is good 期望dp

f[i][j]表示有i张红j张黑情况下的期望收益

f[i][j]=max(0,i/(i+j)*(f[i-1][j]+1)+j/(i+j)*(f[i][j-1]-1))

转移是和很好理解的,如果选了的期望收益<0,那么不如不选

注:

1、直接开数组会爆内存,所以要用滚动数组

2、注意不要四舍五入


#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<iostream>
#include<algorithm>
#define maxn 5010

using namespace std;

double f[2][maxn];
int n,m;

double cal(double x)
{
	if (x<0) return 0;
	else return x;
}

int main()
{
	scanf("%d%d",&n,&m);
	for (int i=0,now=0;i<=n;i++,now^=1)
	{
		f[now][0]=i;
	  	for (int j=1;j<=m;j++)
	      f[now][j]=cal(i/(double)(i+j)*(f[now^1][j]+1)+j/(double)(i+j)*(f[now][j-1]-1));
	}
	long long ans=floor(f[n&1][m]*1000000);
	printf("%lld.%06lld",ans/1000000,ans%1000000);
	return 0;
}


你可能感兴趣的:(【bzoj1419】Red is good 期望dp)