P1168 中位数 - 堆

对顶堆,维护第k大
考虑维护一个大根堆,一个小根堆。
这两个堆总大小为当前区间长度,我们维护两个性质:

1.小根堆总比大根堆多一个元素
2.小根堆所有元素都大于大根堆元素

这样小根堆的堆顶就是区间中位数。

当需要加入一个新元素时,如果这个元素比小根堆堆顶要大,就加入小根堆中,反之则加入大根堆,这样可以维护性质2

加入元素后,检查两个堆的大小关系,如果大小关系不满足性质1,就在两个堆之间搬运元素,由于加入规则的限制我们这样搬来搬去仍然是符合性质2的

#include <algorithm>
#include <iostream>
#include <cstdio>
#include <queue>
using namespace std;
#define debug(x) cerr << #x << "=" << x << endl;
priority_queue <int, vector<int>, greater<int> > qmin;
priority_queue <int> qmax;
int n,a[100010];
int main() {
	cin >> n;
	for(int i=1; i<=n ;i++) {
		cin >> a[i];
		if(qmin.empty()) {
			qmin.push(a[i]);
			cout << a[i] << endl;
			continue;
		}
		if(a[i] > qmin.top()) 
			qmin.push(a[i]);
		else 
			qmax.push(a[i]);
		while(qmax.size() > qmin.size()-1) {
			qmin.push(qmax.top());
			qmax.pop();
		}
		while(qmin.size() - 1 > qmax.size()) {
			qmax.push(qmin.top());
			qmin.pop();
		}
		if(i % 2)
			cout << qmin.top() << endl;
	}
	return 0;
} 
posted @ 2018-11-02 14:38  Zolrk  阅读(118)  评论(0编辑  收藏  举报