[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);
}