求一个数据流的中位数

217 篇文章 0 订阅
41 篇文章 0 订阅

本题源自剑指offer

----------------------------------------------------------------------

用两个堆来解决这个问题 1用最大堆存储左半部分的数,用最小堆存储有半部分的数据,若有奇位个数,中位数就是最小堆的堆顶,否则为最大堆和最小堆堆顶的平均数。

如果当前数据流中的数为偶数个,则新数据插入最小堆,否则插入最大堆。保证最小堆中的数比最大堆的数大。

    void Insert(int num)
    {
        //总数目为偶数时,插入最小堆,否则插入最大堆,最大堆的数字都比最小堆的小
        if(((max.size()+min.size())&1)==0){
            if(max.size()>0&&max[0]>num){
                max.push_back(num);
                push_heap(max.begin(),max.end(),less<int>());
                num=max[0];
                pop_heap(max.begin(),max.end(),less<int>());
                max.pop_back();
            }
            min.push_back(num);
            push_heap(min.begin(),min.end(),greater<int>());
        }else{
            if(min.size()>0&&num>min[0]){
                min.push_back(num);
                push_heap(min.begin(),min.end(),greater<int>());
                num=min[0];
                pop_heap(min.begin(),min.end(),greater<int>());
                min.pop_back();
            }
            max.push_back(num);
            push_heap(max.begin(),max.end(),less<int>());
        }
    }

    double GetMedian()
    {
        int size=min.size()+max.size();
        if(size==0)
            return -1.0;
        if(size&1){
            return min[0];
        }
        return (double)(min[0]+max[0])/2;
    }


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值