思路:
用一个大顶堆和一个小顶堆,将队列分为两部分,大顶堆的堆顶小于小顶堆的堆顶,这样就连续的把数列分开了,每次询问只要把小顶堆里的数扔到大顶堆里去,然后调整两个堆使得大顶堆的堆顶小于小顶堆的堆顶
,然后此时大顶堆的堆顶就是第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());
}
}