洛谷P1168 中位数 - 堆

这题本来应该是主席树模板的。。。

因为只需要求中位数，并且数据不是特别大，我们可以考虑维护一个大根堆，一个小根堆。

这两个堆总大小为当前区间长度，我们维护两个性质：

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

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

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

加入元素后，检查两个堆的大小关系，如果大小关系不满足性质1，就在两个堆之间搬运元素

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