【剑指offer】数据流中的中位数

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

解析:本题考察大顶堆和小顶堆:
用小顶堆存放后半部分的数(大),大顶堆存放前半部分的数(小),这样两个堆的堆顶就是中间的数字。问题是该如何存放。
数先放到大顶堆,再从大顶堆中弹出,送入小顶堆,这样可以保证大顶堆中的数永远比小堆小。

import java.util.PriorityQueue;
import java.util.Comparator;
public class Solution {
    
    PriorityQueue<Integer> highQueue = new PriorityQueue<>(); // 小顶堆保存大数据部分
    PriorityQueue<Integer> lowQueue = new PriorityQueue<>((Comparator<Integer>) (o1, o2) -> o2 - o1);// 大顶堆保存小数据部分

    public void Insert(Integer num) { // 保持两个堆平衡即size之差为1(默认lowQueue比highQueue >= 1)
        highQueue.offer(num);// 数据先往大堆扔
        lowQueue.offer(highQueue.poll());// 保证存到小堆的永远比大堆中的大。
        if (lowQueue.size() > highQueue.size()) {// 若此时小堆容量大于大堆则移出小堆中最小数据至小堆
            highQueue.offer(lowQueue.poll());
        }
    }

    public Double GetMedian() {
        return lowQueue.size() == highQueue.size() ? ((lowQueue.peek() + highQueue.peek()) * 0.5) : Double.valueOf(highQueue.peek());
    }
    }

这里的Double.valueOf值得关注一下,这里是不管你是字符 如 (‘3’)还是数字,统一会转化为Double的数。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值