天天看点

zoj 3640 Help Me Escape 记忆化搜索

     山洞有N个出口,每个出口都有一个难度c[i],如果当前的战斗力大于这条路的难度,就可以花t[i]天走出山洞,否则就会在这条路上被困1天,并且战斗力增加c[i],给出初始的战斗力求走出山洞期望的天数。

dp[k]记录战斗力为k时走出山洞的概率,求期望时遍历每个山洞,如果k>c[i],那么dp[k]+=t[i]/n;否则dp[k]+=(1+dp[c[i]+k])/n。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
using namespace std;
typedef long long ll;
const double g=1.618033988749;
double dp[100000];
const double eps=0.0000000001;
int n,m,f;
int t[220];
int c[220];
int maxn;
double find(int k)
{
    if (fabs(dp[k])>eps) return dp[k];
    for (int i=1; i<=n; i++)
    {
       if (k>c[i]) dp[k]+=(double)t[i]/(double)n;
       else dp[k]+=(double)(1.0+find(k+c[i]))/(double)n;
    }
    return dp[k];
}
int main()
{
//    freopen("in.txt","r",stdin);
    while(~scanf("%d%d",&n,&f))
    {
        memset(dp,0,sizeof dp);
        maxn=0;
        for (int i=1; i<=n; i++)
        {
            scanf("%d",&c[i]);
            maxn=max(maxn,c[i]);
            t[i]=(g*(double)(c[i]*c[i]));
        }
        double ans=0.0;
        ans=find(f);
        printf("%.3lf\n",ans);
    }
    return 0;
}