poj 2431 解题记录

题意

给出加油站的位置和油量，以及目前卡车的位置和现有油量，假设卡车的油箱无限大，问最少要加多少次油可以到达终点？

思路

要让我来想这道题估计是想不出来的，我是看了《挑战程序设计》这本书，里面提到了这道题的思路。思路需要做一个转换，本来是先决定加不加油，然后导致结果油够不够用，这种思路要转换成看有够不够用，来决定之前加不加油。。要直接想还有点难度

代码

我写了两个练手，一个是用的stl的优先队列，另外是自己手写了个优先队列，时间复杂度都是O（nlogn）（应该没算错吧。。没把握）

stl的优先队列实现

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
struct Node
{
    int dis,fuel;
    Node(int a,int b):dis(a),fuel(b){};
};
class comp1
{
    public:
        bool operator()(const Node& a,const Node& b)
        {
            return a.dis<b.dis;
        }
};
class comp2
{
    public:
        bool operator()(const Node& a,const Node& b)
        {
            return a.fuel<b.fuel;
        }
};
vector<Node> v;
int main()
{
    int n;
    cin>>n;
    int i;
    int l,p;
    v.push_back(Node(0,0));
    for(i=0;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        v.push_back(Node(a,b));
    }
    cin>>l>>p;
    sort(v.begin(),v.end(),comp1());
    priority_queue<Node,vector<Node>,comp2> pq;
    i = v.size()-1;
    int cnt = 0;
    while(i>=0)
    {
        int d = l - v[i].dis;
        p -= d;
        l = v[i].dis;
        if(p<0)
        {
            while(!pq.empty()&&p<0)
            {
                Node t = pq.top();
                pq.pop();
                p += t.fuel;
                cnt++;
                //cout<<t.dis<<endl;
            }
            if(p<0)
            {
                cout<<"-1"<<endl;
                return 0;
            }
        }
        pq.push(v[i]);
        i--;
    }
    cout<<cnt<<endl;
    return 0;
}

结果：Memory: 436K Time: 125MS

自己写的优先队列

#include <iostream>
#include <algorithm>
#include <queue>
#include <vector>
using namespace std;
struct Node
{
    int dis,fuel;
    Node(int a=0,int b=0):dis(a),fuel(b){};
};
Node heap[10000+10];
int sz = 0;
void push(Node& x)
{
    int i = sz;
    heap[sz++] = x;
    int p = (i-1)/2;
    if(i==0)
        return;
    while(i!=0&&heap[p].fuel<x.fuel)
    {
        heap[i] = heap[p];
        i = p;
        p = (i-1)/2;
    }
    heap[i] = x;
}
Node pop()
{
    int i = 0;
    Node ret = heap[0];
    Node x = heap[--sz];
    while(2*i+1<sz)
    {
        int a = 2*i+1;
        int b = a+1;
        if(b<sz&&heap[b].fuel>heap[a].fuel)
            a = b;
        if(heap[a].fuel>x.fuel)
        {
            heap[i] = heap[a];
            i = a;
        }
        else
            break;
    }
    heap[i] = x;
    return ret;
}
class comp1
{
    public:
        bool operator()(const Node& a,const Node& b)
        {
            return a.dis<b.dis;
        }
};
class comp2
{
    public:
        bool operator()(const Node& a,const Node& b)
        {
            return a.fuel<b.fuel;
        }
};
vector<Node> v;
int main()
{
    int n;
    cin>>n;
    int i;
    int l,p;
    v.push_back(Node(0,0));
    for(i=0;i<n;i++)
    {
        int a,b;
        cin>>a>>b;
        v.push_back(Node(a,b));
    }
    cin>>l>>p;
    sort(v.begin(),v.end(),comp1());
    i = v.size()-1;
    int cnt = 0;
    while(i>=0)
    {
        int d = l - v[i].dis;
        p -= d;
        l = v[i].dis;
        if(p<0)
        {
            while(sz>0&&p<0)
            {
                Node t = pop();
                p += t.fuel;
                cnt++;
                //cout<<t.dis<<endl;
            }
            if(p<0)
            {
                cout<<"-1"<<endl;
                return 0;
            }
        }
        push(v[i]);
        i--;
    }
    cout<<cnt<<endl;
    return 0;
}

结果：Memory: 520K Time: 172MS
所以标准库还是很给力的，还有就是可能我自己的优先队列实现太渣了。。