【CF865C】Gotta Go Fast 二分+期望DP

【CF865C】Gotta Go Fast

题意:有n个关卡需要依次通过,第i关有pi的概率要花ai时间通过,有1-pi的概率要花bi时间通过,你的目标是花费不超过m的时间通关,每一关开始时你都可以选择进行这个关卡或是重新开始。问你达成目标的最短期望总时间(假设你是绝顶聪明的)。

n<=50,m<=5000。

题解:设f[i][j]表示已经完成了前i关,用了j的时间,期望的通关最小总时间。那么每一关开始时你都可以选择打或不打,所以得到DP方程:

f[i-1][j]=min(f[0][0],(f[j+ai]+ai)*pi+(f[j+bi]+bi)*(1-pi))

但是问题来了,我们现在还不知道f[0][0]的值,怎么办?

我们可以二分!容易发现,当我们假定的f[0][0]比真正值大时,算出来的f[0][0]要比假定值小,反之比假定值大。这个原理感觉和分数规划的思想差不多。

#include <cstdio>
#include <cstring>
#include <iostream>
using namespace std;
double f[55][5110];
int n,m;
int a[55],b[55];
double p[55];
bool check(double mid)
{
        int i,j;
        for(i=0;i<=n;i++)    for(j=0;j<=m+100;j++)        f[i][j]=mid;
        for(j=0;j<=m;j++)    f[n][j]=0;
        for(i=n;i>=1;i--)
        {
                for(j=0;j<=m;j++)
                {
                        f[i-1][j]=min(mid,(f[i][j+a[i]]+a[i])*p[i]+(f[i][j+b[i]]+b[i])*(1-p[i]));
                }
        }
        return f[0][0]<mid;
}
inline int rd()
{
        int ret=0,f=1;  char gc=getchar();
        while(gc<\'0\'||gc>\'9\') {if(gc==\'-\')  f=-f;   gc=getchar();}
        while(gc>=\'0\'&&gc<=\'9\')       ret=ret*10+(gc^\'0\'),gc=getchar();
        return ret*f;
}
int main()
{
        n=rd(),m=rd();
        int i;
        double l=0,r=34751706,mid;
        for(i=1;i<=n;i++)    a[i]=rd(),b[i]=rd(),p[i]=rd()*0.01;
        for(i=1;i<=60;i++)
        {
                mid=(l+r)/2;
                if(check(mid))  r=mid;
                else    l=mid;
        }
        printf("%.9f",r);
        return 0;
}