题目大意:有一个吸血鬼,初始攻击力为f,每天随机走到n个洞里面,每个洞有一个c[i],如果他的攻击力f>c[i]
则可以花费t[i] 的时间逃走,否则则花费一天时间使自己的攻击力增加c[i],求逃走天数的期望
分析:
这道题求期望,,考虑采用概率dp求解
想到的最简单方法就是dp[i][j]表示 第i天,攻击力为j的概率,然后对每一个c进行转移,最后统计答案
但是发现i,j的范围都是10000,n是100 这么做显然是行不通的
于是又可耻的搜了一下题解,发现有一个博主写的期望dp这个概念很不错
令 dp[a]表示 攻击力为 a 后 还需要多少天逃出的期望,那么dp[f]即为答案
注意关键是这个 “还”
状态转移:
如果 a>c[i] 那么显然还需要 t[i]时间逃出
如果 a<=c[i] 那么先要花费一天把攻击力增加a+c[i],然后还要花费的时间就是 dp[a+c[i]]
这样转移方程就很好写了
由于需要从后往前转移,采用记忆化搜索写
开始数组开10010 老是segment fault 后来开到10W过了。。
代码:
#include <iostream>
#include <stdio.h>
#include<string.h>
#include<algorithm>
#include<string>
#include<ctype.h>
#include<math.h>
#pragma comment(linker, "/STACK:102400000,102400000")
using namespace std;
int n,f;
double p;
double dp[];
int c[];
bool vi[];
double dfs(int a)
{
if(vi[a])
return dp[a];
vi[a]=;
for(int i=;i<n;i++)
{
if(a>c[i])
dp[a]+=(double)floor(p*c[i]*c[i])/(double)n;
else
dp[a]+=(1.0+dfs(a+c[i]))/(double)n;
}
return dp[a];
}
int main()
{
while(scanf("%d%d",&n,&f)!=EOF)
{
for(int i=;i<n;i++)
{
scanf("%d",c+i);
}
memset(dp,,sizeof(dp));
memset(vi,,sizeof(vi));
p=(1.0+sqrt(5.0))/2.0;
printf("%.3f\n",dfs(f));
}
return ;
}