POJ 2431 Expedition（优先队列、贪心）

题目链接: 传送门

Expedition

Time Limit: 1000MS     Memory Limit: 65536K

题目描述

驾驶一辆卡车行驶L单位距离。最开始有P单位的汽油。卡车每开1单位距离消耗1单位的汽油。在途中一共有N个加油站。假设卡车的燃料箱容量无限大，问如果到达终点最少的加油次数。

思路

在卡车开往终点途中，只有在加油站才可以加油，换做认为“在到达加油站i时，就获得了一次在之后任何时候都可以加油的权利”

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
struct oil{
	int dis,fuel;
};
bool cmp(struct oil x,struct oil y)
{
	return x.dis <= y.dis;
}

int main()
{
	int N;
	
	while (scanf("%d",&N) != EOF)
	{
		struct oil stops[10005];
		priority_queue<int>pque;
		int L,P;
		
		for (int i = 0;i < N;i++)
		{
			scanf("%d%d",&stops[i].dis,&stops[i].fuel);
		}
		
		scanf("%d%d",&L,&P);
		
		for (int i = 0;i < N;i++)
		{
			stops[i].dis = L - stops[i].dis;
		}
		
		sort(stops,stops + N,cmp);
		
		int cnt = 0;
		bool flag = true;
		for (int i = 0;i < N;i++)
		{
			if (P >= L)
			{
				break;
			}
			
			if (P < stops[i].dis)
			{
				if (pque.empty())
				{
					flag = false;
					break;
				}
				else
				{
					while (!pque.empty() && P < stops[i].dis)
					{
						cnt++;
						int tmp = pque.top();
						pque.pop();
						P += tmp;
					}
					if (pque.empty() &&  P < stops[i].dis)
					{
						flag = false;
						break;
					}
					pque.push(stops[i].fuel);
				}
			}
			else
			{
				pque.push(stops[i].fuel);
			}
		}
		
		if (P < L && !pque.empty())
		{
			int tmp = pque.top();
			pque.pop();
			P += tmp;
			cnt++;
		}
		
		if (!flag)
		{
			printf("-1\n");
		}
		else
		{
			printf("%d\n",cnt);
		}
	}
	return 0;
}