您的位置:首页 > 其它

poj 1442 优先队列

2016-07-12 16:48 183 查看
题意:

每次插入一个数字,在第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
,u
,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;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: