加油站问题 (优先队列)

输入:

N=4,L=25,P=10

A={10,14,20,21}

B={10,5,2,4}

输出:

2  (在第1个和第2个加油站加油)

1.如果优先队列也是空的,则无法到达终点.

2.否则取出优先队列中的最大元素,并用来给卡车加油.

int L,P,N;
int A[MAX],B[MAX];

void solve()
{
    //为了方便,我们把终点也认为是加油站
    A[N]=L;
    B[N]=0;
    N++;
    //维护加油站的优先队列
    priority_queue<int> q;
    //ans:加油次数,pos:现在的位置,tank油箱中汽油的量
    int ans=0,pos=0,tank=P;

    for(int i=0; i<N; i++){
        //接下去要前进的距离
        int d=tank-pos;
        //不断加油直到油量足够行驶到下一个加油站
        while(tank-d<0){
            if(q.empty()){
                puts("-1");
                return ;
            }
            tank+=q.top();
            q.pop();
            ans++;
        }

        tank-=d;
        pos=A[i];
        q.push(B[i]);
    }
    printf("%d
",ans);
}