堆

数据流的中位数

困难#295Java时间 插入 O(log n),查询 O(1)空间 O(n)

题目要做什么

设计数据流中位数结构。addNum 插入一个整数,findMedian 返回当前中位数,查询时数据流必须非空。

题目示例输入ops = ["MedianFinder","addNum","addNum","findMedian","addNum","findMedian"], args = [[],[1],[2],[],[3],[]]
题目示例输出[null,null,null,1.5,null,2]

01 为什么这样做

左堆保存较小一半,用大根堆暴露其最大值;右堆保存较大一半,用小根堆暴露其最小值。两堆等大或左边多一个,根节点即可决定中位数。

始终成立的条件

左堆所有值 ≤ 右堆所有值;左堆大小等于右堆或多一。

02 看见算法执行

动画与代码同步

修改输入,播放自己的例子

当前使用题目示例
修改输入会改变执行过程;预期输出用于核对结果。
数组 / nums步骤 1
执行状态
本次演示输出播放到最后查看

Solution.java参考解法
1class MedianFinder {2    private final PriorityQueue<Integer> left = new PriorityQueue<>(Comparator.reverseOrder());3    private final PriorityQueue<Integer> right = new PriorityQueue<>();4    public MedianFinder() { }5    public void addNum(int num) {6        left.add(num);7        right.add(left.remove());8        if (left.size() < right.size()) left.add(right.remove());9    }10    public double findMedian() {11        if (left.size() > right.size()) return left.peek();12        return ((long) left.peek() + right.peek()) / 2.0;13    }14}

键盘控制:A 后退 · D 前进

编程练习

↑↓ 选择↵ 打开Pagefind 全文检索