63、数据流中的中位数

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

代码

//大顶堆存小值，从大到小排；小顶堆存大值，从小到大排。两个堆顶的元素就是中间的元素
//比如：321,456->两个堆顶元素是3和4；
import java.util.PriorityQueue;
import java.util.Comparator;
public class Solution {
    //定义小顶堆，PriorityQueue默认是从小到大排序
    private PriorityQueue<Integer> minHeap = new PriorityQueue<>();
    //定义大顶堆，15是初始容量，用比较器排序，改成从大到小排序
    private PriorityQueue<Integer> maxHeap = new PriorityQueue<>(15,new Comparator<Integer>(){
        public int compare(Integer o1,Integer o2){
            return o2 - o1;
        }
    });
    int count = 0;//记录奇偶数，其实insert的时候是部分奇偶的，都会加到大顶堆或者小顶堆然后poll出来加到另外一个
    public void Insert(Integer num) {
        if(count % 2 == 0){
            //个数为偶数的话，先插入到大顶堆，再把大顶堆中的最大值拿出来放到小顶堆中
            maxHeap.offer(num);
            int max = maxHeap.poll();
            minHeap.offer(max);
        }else{
            //个数为奇数的话，先插入到小顶堆，再把小顶堆中的最小值拿出来放到大顶堆中
            minHeap.offer(num);
            int min = minHeap.poll();
            maxHeap.offer(min);//maxHeap重写了排序方法，所以offer进来自动排序
        }
        count++;
    }

    public Double GetMedian() {
        //个数为偶数，取小顶堆和大顶堆的堆顶元素求平均值
        if(count % 2 == 0){
            //不能是return new Double((maxHeap.peek() + minHeap.peek()) / 2);
            //上边是对int/2后再取Double，在int/2的过程中小数除没啦，所以先Double再/2
            return new Double(maxHeap.peek() + minHeap.peek()) / 2;
        }else{
            return new Double(minHeap.peek());
        }
    }


}