如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。

题目描述

如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。

/*
思路:
构建一个最大堆和最小堆,分别保存一半的数据,当为奇数时最大堆比最小堆多1;
构建计数器:
当为奇数时,数据先入最小堆,再将最小堆堆顶元素放入最大堆堆顶。(相当于最大堆元素加1)
当为偶数时,数据先入最大堆,再将最大堆堆顶元素放入最小堆堆顶。(相当于最小堆元素加1)
这样使得最大堆得所以元素都小于最小堆所以元素
*/

class Solution {
public:
    void Insert(int num)
    {
        count += 1;
        if(count%2==0){
            big_heap.push(num);
            small_heap.push(big_heap.top());
            big_heap.pop();
        }
        else{
            small_heap.push(num);
            big_heap.push(small_heap.top());
            small_heap.pop();
        }
    }

    double GetMedian()
    { 
        if(count%2==0){
            return double((small_heap.top() + big_heap.top())/2.0);
        }else{
            return big_heap.top();
        }
    }
private:
    int count = 0;
    priority_queue<int,vector<int>,less<int>> big_heap;//最大堆
    priority_queue<int,vector<int>,greater<int>> small_heap;//最小堆

};

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值