洛谷P1168 中位数

来源:互联网 发布:淘宝客服不回 编辑:程序博客网 时间:2024/05/17 07:31

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

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

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

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;} 
原创粉丝点击