poj 1442 Black Box

/*

题意:向一个容器中不断的添加m个数,然后有n个查询,
每次查询在插入的前u个数中第i个小的数,i初始值是1,每查询一次就自增1
sample:        7 4            3 1 -4 2 8 -1000 2                1 2 6 6
第1次查询:u=1,即插入 3 , 第1 小的数是 3
第2次查询:u=2,即插入 3 1, 第2 小的数是 3
第3次查询:u=6,即插入 3 1 -4 2 8 -1000, 第3 小的数是 1
第3次查询:u=6,即插入 3 1 -4 2 8 -1000, 第4 小的数是 2
所以输出为: 3 3 1 2
思路: 要查询第 i 大的数,需要维护大顶堆和小顶堆,让大顶堆的堆顶小于小顶堆的堆顶
即大顶堆所有元素都小于小顶堆的所有元素.每次插入新元素时,让大顶堆的规模保持为 i-1
则小顶堆的堆顶就是第 i 小的数了

*/
#include <iostream>        //大顶堆和小顶堆,查询第i小的数
#include <queue>
using namespace std;
int m,n,arr[30002],res[30002];
int main()
{
priority_queue<int> q1;    //大顶堆
priority_queue<int,vector<int>,greater<int> > q2;    //小顶堆
cin>>m>>n;
for(int i=1;i<=m;++i)
cin>>arr[i];
int u,cur=0,cnt=0,rear=0;
while(n--)
{
cin>>u;
while(cur<u)
{
int num=arr[++cur];        // cur 指当前处理的arr数组的下标
//以下两种操作都是让大顶堆的规模保持不变
if( !q1.empty() && num<q1.top() )    // 新元素小于大顶堆q1的最大值(堆顶),则需要先弹出q1的堆顶,并把它压入小顶堆,再把新元素压入到大顶堆
{
int t=q1.top();
q1.pop();
q2.push(t);
q1.push(num);
}
else    // 新元素比大顶堆q1的最大值(堆顶)还大,则压入小顶堆
{
q2.push(num);
}
}
if(q1.size()<cnt)    // cnt 指大顶堆要达到的规模
{
int t=q2.top();
q2.pop();
q1.push(t);
}
res[rear++]=q2.top();    //小顶堆的堆顶就是第cnt+1小的数
cnt++;
}
for(int i=0;i<rear;++i)
cout<<res[i]<<endl;
return 0;
}