63. 数据流的中位数

题目描述

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

Solution

使用一个小顶堆和一个大顶堆存储数据，小顶堆所有的数都大于大顶堆，插入数据时保持两个堆大小相差不超过1，这样中位数会在小顶堆和大顶堆的堆顶产生

1. 当前数据量为偶数时，插入新数据时，要让大顶堆大小加1，将新数据插入小顶堆中，将小顶堆堆顶移入大顶堆（按序插入新数据）
2. 当前数据量为奇数时，插入新数据时，要让小顶堆大小加1，保持两个堆大小相差不超过1，将新数据插入大顶堆中，将大顶堆堆顶移入小顶堆（按序插入新数据）
3. 当前数据量为偶数时，中位数为两个堆顶的平均值；当前数据量为奇数时，中位数为大顶堆堆顶

import java.util.*;
public class Solution {

    private PriorityQueue<Integer> maxHeap = new PriorityQueue<>((o1, o2) -> o2 - o1);
    private PriorityQueue<Integer> minHeap = new PriorityQueue<>();
    private int n = 0;
       
    public void Insert(Integer num) {
        if (n % 2 == 0) {
            minHeap.offer(num);
            maxHeap.offer(minHeap.poll());
        } else {
            maxHeap.offer(num);
            minHeap.offer(maxHeap.poll());
        }
        n++;
    }

    public Double GetMedian() {
        if (n % 2 == 0) {
            return (maxHeap.peek()+minHeap.peek())/2.0;
        } else {
            return (double) maxHeap.peek();
        }
    }
}