優先隊列（堆）經典例題——poj1442 black box

題意

　兩種操作，第一種可以向序列里加數，第二種查詢長度為k時的序列的第i大元素。

思路   

　優先隊列維護最大堆和最小堆分別存放前i-1大的元素前k-i小的元素。

　將當前序列的元素壓入最小堆，如果最小堆的最小數大於最大堆的最大數則進行交換，保證最大堆中的所有數小於最小堆。

　因為i值每進行一次自增1，所以每次GET操作后將最小堆中的最小數彈出存入最大堆。

易錯點

　Get()的While循環必須放在插入之后進行判斷，否則若放在插入之前寫作 while (j<N && u[j]==i)，當不再插入A(n)時，剩下的Get()將不再執行，導致WA。

代碼

#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int MAXN=30000;
int A[MAXN];
int u[MAXN];
int M,N;
 
int main()
{
    scanf("%d%d",&M,&N);
    for (int i=0;i<M;i++) scanf("%d",&A[i]);
    for (int i=0;i<N;i++) scanf("%d",&u[i]);
    priority_queue< int,vector<int>,less<int> > maxHeap;//大根堆1--（i-1） 
    priority_queue< int,vector<int>,greater<int> > minHeap;//小根堆i--k 
    int j=0;
    for (int i=0;i<M;i++)
    {
        minHeap.push(A[i]);//向小根堆插入 
        if (!maxHeap.empty() && maxHeap.top()>minHeap.top())//大根堆不為空且大根堆堆頂大於小根堆堆頂時，交換 
        {
            int temp1=maxHeap.top();maxHeap.pop();
            int temp2=minHeap.top();minHeap.pop();
            maxHeap.push(temp2);
            minHeap.push(temp1);
        }
        while (j<N && u[j]==i+1)//此時長度剛好 
        {
            cout<<minHeap.top()<<endl;
            maxHeap.push(minHeap.top());
            minHeap.pop();
            j++;
        }
    }
    return 0;
}