[Codeforces 865C]Gotta Go Fast，期望dp+二分答案

[Codeforces 865C]Gotta Go Fast(期望dp+二分答案)

题面

一个游戏一共有n个关卡，对于第i关，用a[i]时间通过的概率为p[i],用b[i]通过的时间为1-p[i]，每通过一关后可以选择继续下一关或者时间清0并从第一关开始，先要求通过所有关卡的时间和不能超过R才算彻底通关，问直到彻底通关位置的游戏时间的期望值为多少

分析

二分从头开始通关的用时期望mid

设\(dp[i][j]\)表示通前i关，当前时间为j的期望，倒推期望.

若超时重新开始，则\(dp[i][j]=mid\)

若用方法a通过这一关，则更新j，\((dp[i+1][j+a[i]]+a[i])*p[i]\)

用方法b同理

总而言之，有$$dp[i][j]=min(mid,(dp[i+1][j+a[i]]+a[i])p[i],(dp[i+1][j+b[i]]+b[i])(1-p[i]))$$

如果最终答案\(dp[1][0]<mid\)，就缩小二分范围，否则增大二分范围

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxn 50
#define maxv 5000
#define maxrb 1e10
#define maxt 150 //二分答案迭代次数，处理精度 
using namespace std;
int n,r;
int a[maxn+5],b[maxn+5];
double p[maxn+5];
double dp[maxn+5][maxv+5];
//dp[i][j]第i关，用时j通关的期望
//二分从1关开始通完的期望mid 
bool check(double mid){
        for(int i=n;i>=1;i--){
                for(int j=r+1;j<=maxv;j++){
                        dp[i+1][j]=mid;
                        //通关超过时间限制，回到起点重新开始 
                }
                for(int j=0;j<=r;j++){
                        double t1=(dp[i+1][j+a[i]]+a[i])*p[i];
                        double t2=(dp[i+1][j+b[i]]+b[i])*(1-p[i]);
                        dp[i][j]=min(mid,t1+t2);
                }
        }
        if(dp[1][0]<mid) return 1;//如果实际dp值比二分值更小，可以继续缩小二分范围 
        else return 0;
} 
int main(){
        scanf("%d %d",&n,&r);
        for(int i=1;i<=n;i++){
                scanf("%d %d",&a[i],&b[i]);
                scanf("%lf",&p[i]);
                p[i]/=100;
        }
        double lb=0,rb=1e10,mid,ans=0;
        for(int i=1;i<=maxt;i++){
                mid=(lb+rb)/2;
                if(check(mid)){
                        ans=mid;
                        rb=mid;
                }else lb=mid;
        }
        printf("%.9lf",ans);
}