P1801 黑匣子题解 对顶堆

题意大致是这样的：

有两种指令，分别是“添加”和“查找第i大的数”，其中i初始为1，每次查找后使i+1

维护两个堆。

大根堆负责存储第i大之前的数字。

小根堆负责存储包含第i大及之后的数字。

大根堆的大小为i-1

当一个数字可以被大根堆容下的时候就存入大根堆，否则将大根堆堆顶放入小根堆。

查询的时候输出小根堆的堆顶。

然后别忘了查询完后将小根堆的堆顶存入大根堆，然后再弹出。（因为查询后i+1，第i大的就变为小根堆中第二小的数）

代码如下

#include<bits/stdc++.h>
using namespace std;
priority_queue<int> zq;
priority_queue<int,vector<int>,greater<int>> wq;
int n,m,a[200010],u;
int main(){
	int cnt=1;
	cin>>m>>n;
	for(int i=1;i<=m;i++){
		cin>>a[i];
	}
	for(int i=1;i<=n;i++){
		cin>>u;
		for(int j=cnt;j<=u;j++){
			zq.push(a[j]);
			if(zq.size() == i){
				wq.push(zq.top());
				zq.pop();
			}
		}
		cnt=u+1;
		printf("%d\n",wq.top());
		zq.push(wq.top());
		wq.pop();
	}
	return 0;
}