PAT 1033. To Fill or Not to Fill (25)

1033. To Fill or Not to Fill (25)

开始用dp写，只过了2个，真心蛋疼。网上看了是贪心，我的思路是：

当前在第i个位置，gas表示剩余汽油能走的距离

找到下个能到达的比第i个更便宜的位置记为j，没有则是-1

1.如果找不到这样的j（即为-1），在i加满油，去下一站

2.如果剩余gas能到j，则去j站

3.如果加上c的gas刚好能到j，则去j站

否则就是不能到，最远距离为i站距离+剩余gas

第3个case是0位置没有加油站的，要注意。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;

const int N=505,inf=1e9;
int C,D,avg,n;
struct node{
	double p;
	int d;
}s[N];
int cmp(node a,node b){
	return a.d<b.d;
}
int find(int b,int e,int cur){//找到下一站更便宜的
	double price=s[cur].p;
	for(int i=cur+1;i<=n;i++){
		if(s[i].d>e) break;
		if(price>s[i].p)
			return i;
	}
	return -1;
}
int main(){
	cin>>C>>D>>avg>>n;
	s[0].d=D,s[0].p=-1;
	for(int i=1;i<=n;i++)
		scanf("%lf %d",&s[i].p,&s[i].d);
	sort(s,s+n+1,cmp);
	int cur=0,gas=0,far=C*avg;
	double ans=0;
	if(s[cur].d>0){
		printf("The maximum travel distance = %.2lf\n",0);
		return 0;
	}
	while(s[cur].d<D){
		if(s[cur].d+far<s[cur+1].d){
			printf("The maximum travel distance = %.2lf\n",1.0*(s[cur].d+far));
			return 0;
		}
		int next=find(s[cur].d,s[cur].d+gas,cur);
		if(next!=-1){//已经够下一站
			gas-=s[next].d-s[cur].d;
			cur=next;
			continue;
		}
		next=find(s[cur].d,s[cur].d+far,cur);
		if(next==-1){//没有更便宜的 加满
			ans+=1.0*(far-gas)/avg*s[cur].p;
			gas=far-(s[cur+1].d-s[cur].d);
			cur=cur+1;
		}
		else{//更便宜的是next
			ans+=1.0*(s[next].d-s[cur].d-gas)/avg*s[cur].p;
			gas=0;
			cur=next;
		}
	}
	printf("%.2lf\n",ans);
	return 0;
}