LeetCode刷题笔记 295 / 剑指Offer 41（涉及到大小顶堆）

题目：数据流的中位数

答案：

参考链接：暴力法+二分查找+手动实现大小堆

1.排序

每次新进来一个数，都把它与已存的数进行排序。

可以选择插入排序，将数字插入到已排好序的序列中，并保持整个序列的排序状态。
可以通过二分搜索，查找到正确位置，然后将元素插入到正确位置。

2.大顶堆+小顶堆

维护两个堆：
用于存储输入数字中较小一半的大顶堆 lo
用于存储输入数字的较大一半的小顶堆 hi

需要保证这 2 个堆的“平衡”。这里的平衡指得是：大顶堆的大小 = 小顶堆的大小， 或者 大顶堆的大小 = 小顶堆的大小 + 1。

当查询中位数的时候，中位数就是最大堆的堆顶元素，或者 (最大堆的堆顶元素 + 最小堆的堆顶元素)/2

剩下的问题就是怎么保证堆的平衡？步骤如下：

（1）先让 num 入大顶堆
（2）取出大顶堆的堆顶元素，放入小顶堆
（3）若此时最大堆的大小 < 最小堆的大小，取出小顶堆的堆顶元素，让入大顶堆

class MedianFinder {
    PriorityQueue<Integer> hi = new PriorityQueue();   //小顶堆
    PriorityQueue<Integer> lo = new PriorityQueue(Collections.reverseOrder()); //大顶堆

    /** initialize your data structure here. */
    public MedianFinder() {
        
    }
    
    public void addNum(int num) {
        lo.add(num);
        hi.add(lo.poll());
        
        if(lo.size() < hi.size()){
            lo.add(hi.poll());
        }
    }
    
    public double findMedian() {
        return lo.size() > hi.size() ? (double)lo.peek() : (lo.peek() + hi.peek()) * 0.5;
    }
}