堆/优先级队列 - 41. 数据流中的中位数

在这里插入图片描述
思路:两个堆,一个最大堆一个最小堆。最大堆保存小的数,最小堆保存大的数。
形象的来个图(帮大家活动下脖子)

在这里插入图片描述
此时我们就可以发现最小堆和最大堆保存的值就是中位数。如果是偶数,直接将两个堆顶加和除二即可。如果是奇数,直接返回最大堆的堆顶即可。
为了让两个堆均衡负载,我们可以通过他们的大小来决定流中的数字进那个堆。

class MedianFinder {
        public  PriorityQueue<Integer> maxHeap,minHeap;
        /** initialize your data structure here. */
        public MedianFinder() {
	        //一个lamda表达式,因为定义的泛型是整形,
	        //所以可以用lamda表大是定义一个比较器。
	        //原本的比较器比较的是a 与 b ,a > b返回1. 也就是 a - b
	        //为了让小根堆变成大根堆,要让他反向比较才可以
	        //所以传入的比较器为 b - a
            maxHeap = new PriorityQueue<Integer>((a,b)->(b - a));
            minHeap = new PriorityQueue<Integer>();
        }

        public void addNum(int num) {
            //直接添加到最小堆中去,如果不平衡添加回来即可。
            maxHeap.offer(num);
            minHeap.offer(maxHeap.poll());
            if(maxHeap.size() < minHeap.size()){
                maxHeap.offer(minHeap.poll());
            }
        }

        public double findMedian() {
            //如果是奇数,直接返回中位数
            //如果是偶数,那么中位数就是最大堆和最小堆顶之和
            if(maxHeap.size() == minHeap.size() ){
                return (maxHeap.peek()+minHeap.peek())*0.5;
            }
            return maxHeap.peek()/1.0;
        }
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值