首页 > 代码库 > zoj-3640-Help Me Escape-概率DP

zoj-3640-Help Me Escape-概率DP

dp[x]:能力为x的时候,逃出的期望。

则,可以根据x,来算出期望的公式。

#include <iostream>
#include<stdio.h>
#include<string.h>
#include<math.h>
using namespace std;
#define maxn 110000
#define eps 1e-6
#define zero(x) (fabs(x)<0?0:x)
double dp[maxn];
int c[maxn];
int t[maxn];
int n,m;
double dos(int x)
{
    if(dp[x]>-0.5)return dp[x];
    dp[x]=0;
    for(int i=1;i<=n;i++)
    {
        if(x>c[i])
        {
            dp[x]+=(1.0/n)*(t[i]);
        }
        else
        {
            dp[x]+=(1.0/n)*(dos(x+c[i])+1);
        }
    }
    return dp[x];
}
int main()
{
    double x,y;
    while(~scanf("%d%d",&n,&m)&&(n||m))
    {
        memset(dp,-1,sizeof(dp));
        x=sqrt(5.0)+1.0;
        x=x/2.0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&c[i]);
            y=1.0*c[i]*c[i];
            t[i]=x*y;
        }
        printf("%.3f\n",dos(m));
    }
    return 0;
}