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

转载于:https://www.cnblogs.com/Zolrk/p/9896160.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值