洛谷P1801黑匣子——对顶堆（大根堆+小根堆）

题目：https://www.luogu.org/problemnew/show/P1801

以样例说明思路：

7 4
3 1 -4 2 8 -1000 2
1 2 6 6

先理解题意：当读入元素分别为1、2、6、6个时，输出第1、2、3、4的元素。

于是想到引入一个小根堆，一个大根堆。保证小根堆的元素都不小于大根堆的元素。大根堆的元素个数每操作1次后容量加1。

手动模拟：

step1

将3存入小根堆p，此时p中的元素为3。将p的根元素3取出并存到大根堆q，此时q的元素为3。输出q的根元素3（不取出）。

step2

将1存入小根堆p，此时p中的元素为1。将p的根元素1取出并存到大根堆q，此时q的元素为3、1。输出q的根元素3（不取出）。

step3

将-4，2，8，-1000依次存入小根堆p，此时p中的元素为-1000，-4，2，8。因为此时p的根元素小于q的根元素，需要交换两者根元素，直到p的根元素不小于q的根元素。

现在p中的元素为1、2、3、8，q的元素为-4、-1000。

将p的根元素1取出并存到大根堆q，输出q的根元素1（不取出）。

现在p中的元素为2、3、8，q的元素为1、-4、-1000。

将p的根元素2取出并存到大根堆q，输出q的根元素2（不取出）。

小结：要保证小根堆的根元素不小于大根堆的根元素。即将序列前i个元素保存在大根堆，将后面的元素保存在小根堆，满足：大根堆的元素都小于（或者不大于）小根堆的元素。

AC代码：

#include<cstdio>
#include<iostream>
#include<queue>
using namespace std;
priority_queue<int,vector<int>,greater<int> >p;//小根堆 
priority_queue<int,vector<int>,less<int> >q;//大根堆
const int MaxN=200005;
int M,N,k,x,a[MaxN],b[MaxN];
int main(){
	cin>>M>>N;
	for(int i=1;i<=M;i++)
		scanf("%d",&a[i]);
	int tot=0,k=0;
	for(int i=1;i<=N;i++){
		scanf("%d",&x);
		while(tot<x){
			p.push( a[++tot] );
		}
		k++;
		int xx=p.top();
		p.pop();
		q.push(xx);
		while(!p.empty() && p.top()<q.top() ){
			//如果p的根小于q的根，则交换，以保证p的元素都大于q的元素 
			int xx=p.top(),yy=q.top();
			q.push(xx);
			q.pop();
			p.pop();
			p.push(yy);
		}
		printf("%d\n",q.top());
	}
	return 0;
}