AcWing 106. 动态中位数 对顶堆 优先队列实现

题目描述
依次读入一个整数序列，每当已经读入的整数个数为奇数时，输出已读入的整数构成的序列的中位数。

输入格式
第一行输入一个整数P，代表后面数据集的个数，接下来若干行输入各个数据集。

每个数据集的第一行首先输入一个代表数据集的编号的整数。

然后输入一个整数M，代表数据集中包含数据的个数，M一定为奇数，数据之间用空格隔开。

数据集的剩余行由数据集的数据构成，每行包含10个数据，最后一行数据量可能少于10个，数据之间用空格隔开。

输出格式
对于每个数据集，第一行输出两个整数，分别代表数据集的编号以及输出中位数的个数（应为数据个数加一的二分之一），数据之间用空格隔开。

数据集的剩余行由输出的中位数构成，每行包含10个数据，最后一行数据量可能少于10个，数据之间用空格隔开。

输出中不应该存在空行。

数据范围
1≤P≤10001≤P≤1000
1≤M≤99991≤M≤9999
样例
输入样例：
3 
1 9 
1 2 3 4 5 6 7 8 9 
2 9 
9 8 7 6 5 4 3 2 1 
3 23 
23 41 13 22 -3 24 -31 -11 -8 -7 
3 5 103 211 -311 -45 -67 -73 -81 -99 
-33 24 56
输出样例：
1 5
1 2 3 4 5
2 5
9 8 7 6 5
3 12
23 23 22 22 13 3 5 5 3 -3 
-7 -3

思路：要求中位数，采用对顶堆的在线做法，读入的同时计算答案。建立一个大根堆和小根堆，序列中从小到大排名1-N/2的放在大根堆中，N/2+1-N的放在小根堆中，则小根堆堆顶元素就是中位数。

代码如下：

//对顶堆算法 
#include <bits/stdc++.h>
using namespace std;
priority_queue<int, vector<int>,greater<int> > q1,p1;//小根堆 
priority_queue<int> q2,p2;//大根堆 
int main(){
	int t,x,n,now;
	cin>>t;
	while(t--){
		cin>>x>>n;
		cout<<x<<" "<<(n+1)/2<<endl;
		q1=p1,q2=p2;
		int cnt=0;
		for(int i=1;i<=n;i++){//序列的中位数是小根堆的堆顶，即q1.top() 
			cin>>now;
			//每次新读入一个数X后，如果X比中位数小，则插入大根堆，否则插入小根堆 
			if(q1.empty()) q1.push(now);
			else{
				if(now>q1.top()){
					q1.push(now);
				}
				else q2.push(now);
				//如果任何一个堆的元素过多，打破了这个平衡，就取出该堆的堆顶放入另一个堆中 
				while(q1.size()<q2.size()){
					q1.push(q2.top());
					q2.pop();
				}
				while(q1.size()>q2.size()+1){
					q2.push(q1.top());
					q1.pop();
				}
			}
			if(i&1){
			    cnt++;//如果是奇数 
				cout<<q1.top()<<" ";//输出中位数 
				if(cnt%10==0) cout<<endl;
			}
		}
		if(cnt!=10) puts("");//防止出现多打换行符的情况
	}
	return 0;
}

(2)vector+upper_bound

#include <bits/stdc++.h>
using namespace std;
int n,x;
vector<int> v;
int main(){
	cin>>n;
	for(int i=1;i<=n;i++){
		cin>>x;
		v.insert(upper_bound(v.begin(),v.end(),x),x);//维护单调性 
		if(i&1) cout<<v[(i-1)/2];//是奇数个就输出 
	}
}