分享一道题目:rollcall

题目描述

初始有一个空集，依次插入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;
}