堆的模板及应用 POJ2431

STL库中可以用优先队列实现堆，以下是自己写的堆及其接口

//Heap
template <typename T> void swap(T &a, T &b){
    T t = a;
    a = b;
    b = t;
}
struct Heap{
    int data[10005];
    int size;
    Heap(){
        for (int i = 0; i < MAXF; ++i)
            data[i] = 0;
        size = 0;
    };
    void upAdjust(int cur){
        if (cur == 1)return;
        if (data[cur] > data[cur / 2]){
            swap(data[cur], data[cur / 2]);
            upAdjust(cur / 2);
        }
    }
    void downAdjust(int cur){
        int max = cur;
        if (2 * cur <= size && data[cur] < data[2 * cur])max = 2 * cur;
        if ((2 * cur + 1) <= size && data[max] < data[2 * cur + 1])
max = 2 * cur + 1;
        if (max != cur){
            swap(data[cur], data[max]);
            downAdjust(max);
        }
    }
    int pop(){
        if (size == 0)return -1;
        int ret = data[1];
        swap(data[1], data[size]);
        if (--size > 1)downAdjust(1);
        return ret;
    }
    int top(){
        return data[1];
    }
    void push(int x){
        ++size;
        data[size] = x;
        upAdjust(size);
    }
    bool empty(){
        return size == 0;
    }
    /*
void build(){
        for (int i = (size - 1) >> 2; i >= 0; --i){
            downAdjust(i);
        }
}
*/
};

练习题：

/*Poj 2431 Heap

题意：一辆卡车要行驶L单位距离。最开始时，卡车上有P单位汽油，每向前行驶1单位距离消耗1单位汽油。如果在途中车上的汽油耗尽，则无法到达终点。途中共有N个加油站，加油站提供的油量有限，卡车的油箱无限大。给出每个加油站距离终点的距离和能够提供的油量，问卡车从起点到终点至少要加几次油？如果不能到达终点，输出-1。

 

分析：因为希望加油次数尽可能少，先假设不需要加油，当行驶到燃料为0了，就会后悔之前路过的加油站，应该找一个油多的站加油。基于贪心的思想，可以用一个优先队列来保存经过的加油站的油量，当需要加油时，取出队列中的最大元素。

*/

#include <stdio.h>
#include <algorithm>
std::pair<int, int> s[MAXF];
int n, l, p;
int main(){
//  freopen("sample_input.txt", "r", stdin);
    while (~scanf("%d", &n)){
        Heap heap;
        heap.init();
        for (int i = 0; i < n; ++i){
            scanf("%d%d", &s[i].first, &s[i].second);
        }
        scanf("%d%d", &l, &p);
        std::sort(s, s + n);
        heap.push(p);
        int cnt = -1, index = n - 1;
        while (l>0 && !heap.empty()){
            int tmp = heap.top();
            heap.pop();
            l -= tmp;
            ++cnt;
            while (index >= 0 && l <= s[index].first){
                heap.push(s[index].second);
                --index;
            }
        }
        printf("%d\n", l <= 0 ? cnt : -1);
    }
    return 0;
}