剑指63 - 数据流中的中位数

剑指63 - 数据流中的中位数

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

思路：
为了保证插入新数据和取中位数的时间效率都高效，这里使用大顶堆+小顶堆的容器，并且满足：
1、两个堆中的数据数目差不能超过1，这样可以使中位数只会出现在两个堆的交接处；
2、大顶堆的所有数据都小于小顶堆，这样就满足了排序要求。

import java.util.*;
public class Solution {
    
    private Queue<Integer> minHeap = new PriorityQueue<>((a1, a2) -> a1-a2);
    private Queue<Integer> maxHeap = new PriorityQueue<>((a1, a2) -> a2-a1);
    private Integer size = 0;

    public void Insert(Integer num) {
        // 小顶堆的数量>=大根堆的数量
        if((size & 1) == 0) {
            maxHeap.offer(num);
            Integer temp = maxHeap.poll();
            minHeap.offer(temp);
        }else{
            minHeap.offer(num);
            Integer temp = minHeap.poll();
            maxHeap.offer(temp);
        }
        size++;
    }

    public Double GetMedian() {
        if((size & 1) == 0) {
            return (minHeap.peek() + maxHeap.peek())*1.0/2;
        } else {
            return minHeap.peek()*1.0;
        }
    }
}