[leetcode] 面试题 17.20. 连续中值 | 对顶堆维护动态中位数

在这里插入图片描述
有很多种形式可以实现中位数的求解,比如将所有的数放到一个数组中,然后sort一下获取中间的值,但这样在时间复杂度上不太优雅;为了能够更快的求解,可以使用对顶堆来求解。
对顶堆通常用来实现动态k大(小)的问题。在这个题里,因为在往里面加数的过程中,数的总数量cnt是在不断加大的,所以说第(cnt+1)/2大也是在不断变化的,正好符合对顶堆的应用场景。

想要了解对顶堆求解第k大可以看博主的另一个博客:传送门

由大根堆和小根堆构成,小根堆在上,大根堆在下,两个堆的根相对,所以说就形成了一个有序的结构(xiao.top() >= da.top())
放数据的时候,往小根堆中添加,然后维护它们之间size()的大小关系即可,比如:
当数的总量为奇数的时候,xiao.size() == da.size() + 1,当数的总量为偶数的时候,xiao.size() == da.size()

还要注意可能会出现xiao.top()<da.top()的情况,比如按照顺序push -1 -2 -3的过程会导致这样的情况,此时还要考虑xiao.top()和da.top()之间的大小关系,然后再处理一下(将大根堆的堆顶元素放入小根堆,然后将小根堆的堆顶元素放到大根堆)

ac_code:

class MedianFinder {
public:
    /** initialize your data structure here. */
    MedianFinder() {
        this->cnt = 0;
    }
    
    void addNum(int num) {
        xiao.push(num);
        cnt ++;
        while(xiao.size() > da.size() + 1) {
            da.push(xiao.top());
            xiao.pop();
        }
        if(xiao.size() && da.size() && (xiao.top() < da.top())) {
            xiao.push(da.top());
            da.pop();
        }
        while(xiao.size() > da.size() + 1) {
            da.push(xiao.top());
            xiao.pop();
        }
    }
    
    double findMedian() {
        // std::cout << cnt << std::endl;
        // std::cout << xiao.top() << " " << xiao.size() << std::endl;
        // std::cout << da.top() << " " << da.size() << std::endl;
        if(xiao.size() == da.size() + 1) return 1.0 * xiao.top();
        else return 1.0 * ((xiao.top() + da.top()) / 2.0);
    }
private:
    std::priority_queue<int,std::vector<int>, std::greater<int> >xiao;
    std::priority_queue<int,std::vector<int>, std::less<int> >da;
    int cnt;
};

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder* obj = new MedianFinder();
 * obj->addNum(num);
 * double param_2 = obj->findMedian();
 */

在这里插入图片描述

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值