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;
} 

猜你喜欢

转载自www.cnblogs.com/Zolrk/p/9896160.html