题目描述
初始有一个空集,依次插入N个数Ai。有M次询问Bj,表示询问第Bj个数加入集合后的排名为j的数是多少
输入格式
第一行是两个整数N,M
接下来一行有N个整数,Ai
接下来一行有M个整数Bj,保证数据合法,Bj单调不减
输出格式
M行,回答每个询问
样例输入
7 4
9 7 2 8 14 1 8
1 2 6 6
样例输出
9
9
7
8
【说明】
第一次询问,当前集合{9},1th=9
第二次询问,当前集合{9,7}的第2=9
第三次询问,当前集合{9,7,2,8,14,1}的第3=7
第四次询问,当前集合{9,7,2,8,14,1}的第4=8
数据规模与约定
【数据规模】
40%的数据保证 n ≤ 1000
100%的数据保证1 ≤ m ≤ n ≤ 30000; 0 ≤ Ai < 2^32
代码实现
#include<bits/stdc++.h>
using namespace std;
const int N=100100;
priority_queue<unsigned,vector<unsigned>,less<unsigned> >A;//记录前i-1小的数(大根堆)
priority_queue<unsigned,vector<unsigned>,greater<unsigned> >B;//记录这Bi个数中剩余的数(小根堆)
unsigned n,m,a[N];
int main(){
ios::sync_with_stdio(0);
cin.tie(0);
cin>>n>>m;
for(int i=1;i<=n;i++)
cin>>a[i];
int cnt=1;
for(int i=1;i<=m;i++){
int q;
cin>>q;
while(cnt<=q){
if(A.empty()||A.top()<a[cnt])//不是前(i-1)小的数
B.push(a[cnt]);
else{//是前(i-1)个数
B.push(A.top());
A.pop();
A.push(a[cnt]);
}
cnt++;
}
cout<<B.top()<<endl;//输出第i小的数
A.push(B.top());//将A中变为前i小的数
B.pop();
}
return 0;
}