您的位置:首页 > 其它

POJ 1442 Black Box(第k次询问第k小,两个堆)

2017-01-13 21:37 169 查看

思路:

用一个大顶堆和一个小顶堆,将队列分为两部分,大顶堆的堆顶小于小顶堆的堆顶,这样就连续的把数列分开了,每次询问只要把小顶堆里的数扔到大顶堆里去,然后调整两个堆使得大顶堆的堆顶小于小顶堆的堆顶

,然后此时大顶堆的堆顶就是第k小。

#include <iostream>
#include <cstdio>
#include <string.h>
#include <queue>
#include <cmath>
typedef long long int lli;
using namespace std;

int a[30100];
int q[30100];

int main(){
int m,n;
cin>>m>>n;
priority_queue<int , vector<int>,greater<int> > q1;
priority_queue<int> q2;

for(int i = 1;i <= m;i++){
scanf("%d",a+i);
}
for(int i = 1;i <= n;i++){
scanf("%d",q+i);
}

int i = 0;
for(int j = 1;j <= n;j++){
while(i < q[j]){
i++;
q1.push(a[i]);
}
int temp = q1.top();
q1.pop();
q2.push(temp);
while(!q1.empty() && q2.top() > q1.top()){
q1.push(q2.top());
q2.pop();
q2.push(q1.top());
q1.pop();
}
printf("%d\n",q2.top());
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: