ZOJ - 3640 Help Me Escape(概率dp)

这题卡时间卡得很紧。

用d[ i ]表示 f = i 时的期望。

if( f > c[ i ] )  d[ i ] += t [ i ] / n;

else d[ i ] += ( 1 + t[ i ] ) / n;

直接从max_c*2开始跑循环正好超时，只能预处理(max_c, max_c*2]的值。

刚好1060ms过。

#include <iostream>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <algorithm>
#include <cstring>
using namespace std;

double d[20010];
int n, f;
int c[110], max_c;
int get_t(int c)
{
    return floor(c*c*(1.0+sqrt(5.0))/2);
}

void dp()
{
    double D=0;
    for(int i=0; i<n; i++)
        D+=get_t(c[i])*1.0/n;
    for(int i=max_c+1; i<=max_c*2; i++)
        d[i]=D;
    for(int i=max(max_c, f); i>=f; i--)
    {
        d[i]=0;
        for(int j=0; j<n; j++)
        {
            if(i>c[j])
                d[i]+=get_t(c[j])*1.0/n;
            else d[i]+=(1+d[i+c[j]])/n;
        }
    }

    printf("%.3lf\n", d[f]);
}

int main()
{
    while(~scanf("%d%d", &n, &f))
    {
        max_c=0;
        for(int i=0; i<n; i++)
        {
            scanf("%d", &c[i]);
            max_c=max(max_c, c[i]);
        }
        dp();
    }
    return 0;
}