力扣-295题 数据流的中位数(C++)- 大小根堆

题目链接:https://leetcode-cn.com/problems/find-median-from-data-stream/
题目如下:
在这里插入图片描述

class MedianFinder {
public:
    MedianFinder() {
        
    }
    
    void addNum(int num) {
        //目的:大根堆中的内容为较小的数,存放的数用于等于或大于小根堆中的数
        if(maxheap.size()==0||num<=maxheap.top()) maxheap.push(num);
        else minheap.push(num);

        //如果大根堆中的数值<小顶堆中的数值,则将小根堆中的数移出,放入大根堆中
        while(maxheap.size()<minheap.size()){
            int e=minheap.top();
            minheap.pop();
            maxheap.push(e);
        }
        //大顶堆最多只能超过小顶堆一个元素,否则,需要移动
        while(minheap.size()<maxheap.size()-1){
            int e=maxheap.top();
            maxheap.pop();
            minheap.push(e);
        }
    }
    
    double findMedian() {
        if(maxheap.size()>minheap.size()) return maxheap.top();
        else return (maxheap.top()+minheap.top())/2.0;
    }
private:
    struct Cmp{
        bool operator()(const int& a,const int& b){
            return a>b;//小根堆
        }
    };
    priority_queue<int,vector<int>,Cmp> minheap;//用于求top K
    priority_queue<int> maxheap;//用于求min K
};

/**
 * Your MedianFinder object will be instantiated and called as such:
 * MedianFinder* obj = new MedianFinder();
 * obj->addNum(num);
 * double param_2 = obj->findMedian();
 */
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值