poj 1442 优先队列

题意:
每次插入一个数字,在第u[i]次插入完后可以查询第k大的数字,然后k递增,初始化k=1。

总共数字n<30000,查询m<=n。

分析:

很容易想到用优先队列去维护已插入的序列有序,然后难处理的就是怎么找到第K大的数字?因为k是递增的,所以每次查询完以后,前k小就不会再出现了,但是可能以后会插入比前k小中某个数还小的,这时候就要想办法把当前插入的数和第k小交换一下。所以可以再利用一个大根堆的优先队列,去保存已经查询过的K个数,然后新插入的数再跟最大数比较,判断是否交换处理即可。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;
const int N=30009;
int a[N],u[N],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>maxq;  //大根堆,保存已查询的前k小的数
    priority_queue<int,vector<int>,greater<int> >minq; //小根堆,队首元素是第k+1小
    int t=0;
    for(int i=0;i<m;i++){
        minq.push(a[i]);
        while(!maxq.empty()&&maxq.top()>minq.top()){
            int maxt=maxq.top();maxq.pop();
            int mint=minq.top();minq.pop();
            maxq.push(mint);
            minq.push(maxt);
        }
        while(t<n&&u[t]==i+1){
            printf("%d\n",minq.top());
            maxq.push(minq.top());
            minq.pop();
            t++;
        }
    }
    return 0;
}


posted @ 2016-07-12 16:48  HARD_UNDERSTAND  阅读(159)  评论(0编辑  收藏  举报