AcWing 106. 动态中位数—对顶堆

chp的博客

已于 2023-07-15 22:18:21 修改

阅读量207

点赞数

分类专栏：数据结构日常训练排序文章标签：算法 c++

于 2023-07-15 17:10:07 首次发布

本文链接：https://blog.csdn.net/qq_43851311/article/details/131740967

版权

日常训练同时被 3 个专栏收录

43 篇文章 0 订阅

订阅专栏

数据结构

35 篇文章 1 订阅

订阅专栏

排序

8 篇文章 0 订阅

订阅专栏

该文介绍了如何利用对顶堆（小根堆和大根堆）来动态维护一个序列的中位数。在C++中，通过STL的优先队列实现堆，保证小根堆的最小元素大于大根堆的最大元素，并保持堆的元素数量差不超过1，从而实时得到中位数。

摘要由CSDN通过智能技术生成

问题链接 AcWing 106. 动态中位数

问题描述
在这里插入图片描述

分析
推荐b站董晓算法视频讲解对顶堆
这道题应该用树状数组、平衡树也能解决，这里用对顶堆来做，对顶堆能够用维护第K位置的数，K是固定的，在这道题中，维护两个堆，一个大根堆一个小根堆
在这里插入图片描述

stl 中有堆，不用自己写，假设a为小根堆，b为大根堆

1.大小关系:
让小根堆中存的元素最小值，大于大根堆元素的最大值，也就是
a.top()>=b.top()

2.数量关系
除了维护大小关系也要维护两个堆内元素的数量关系
b.size()最多比a.size()多1
a.size()最多和b.size()相等
也就是只有两种情况
b.size()-a.size()=1;
b.size()-a.size()=0;

维护好这两堆之后，那么b.top()就是当前序列的中位数

代码如下

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;

int main(){
    int p;
    cin>>p;
    while(p--){
        int k,m;
        cin>>k>>m;
        cout<<k<<" "<<(m+1)/2<<endl;
        priority_queue<int,vector<int>,greater<int> > a;//小根堆
        priority_queue<int,vector<int>,less<int> > b;//大根堆
        for(int i=1;i<=m;i++){
            int x;
            scanf("%d",&x);
            if(!a.empty()&&a.top()<=x) a.push(x);
            else b.push(x);
            if(i%2==1&&b.size()<a.size()){
                b.push(a.top());
                a.pop();
            }
            if(i%2==0&&a.size()<b.size()){
                a.push(b.top());
                b.pop();
            }
            if(i&1) printf("%d ",b.top());
            if(i%20==0) cout<<endl;
        }
        cout<<endl;
    }
    return 0;
}