POJ 2431 Expedition

http://poj.org/problem?id=2431

树上巧妙的思路

每次经过一个stop 就相当于获得一次加油的机会 但是 可以不用这个机会

当没油的时候再加 这个时候可以加的油 最优的方案就是 先按油多的加

优先队列 按按照油降序存储

很像蚂蚁那道题 巧妙的思维啊！

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <queue>
#include <vector>
#include <algorithm>

using namespace std;

struct Stop
{
    int dis, fuel;
    bool operator < (Stop s) const
    {
        return fuel < s.fuel;
    }
}truck;

bool cmp(Stop a, Stop b)
{
    return a.dis > b.dis;
}

priority_queue<Stop> que;
vector<Stop> v;

int N, L, P;
bool flag = true;
int main()
{
    freopen("in.txt", "r", stdin);
   scanf("%d", &N);
   while (N--)
   {
       int dis, fuel;
       Stop tmp;
       scanf("%d%d", &dis, &fuel);
       tmp.dis = dis;
       tmp.fuel = fuel;
       v.push_back(tmp);
   }
   sort(v.begin(), v.end(), cmp);
   scanf("%d%d", &L, &P);
   truck.dis = L;
   truck.fuel = P;
   int j = 0, ans = 0;
   for (int i = 0; i < v.size(); i++)
   if (truck.fuel >= (truck.dis - v[i].dis))
   {
       que.push(v[i]);
   }
   else
   {
       i = i-1;
       if (!que.empty())//如果还有油可以加
       {
            truck.fuel += que.top().fuel;
            que.pop();
            ans++;
       }
       else
       {
           printf("-1
");//这里之前没有直接 退出 而是去设置flag 可能造成了死循环
           return 0;
       }
   }
    while(flag)
    {
        if (truck.dis <= truck.fuel) break;
        else if (!que.empty())
        {
            truck.fuel += que.top().fuel;
            que.pop();
            ans++;
        }
        else flag = false;
    }
   if (flag) printf("%d
", ans);
   else printf("-1
");
   return 0;
}