剑指offer 数据流中的中位数

题目

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

思路
1 利用java的优先队列，分别建立两个堆，大顶堆和小顶堆。
2 因为要求中位数，即我们可以利用大顶堆存储较小的数，小顶堆存放较大的数，这个过程要保证小顶堆里面的数要大于大顶堆里面的数，最后根据堆里面的数字个数来得出中位数。
3 若两个堆里共有偶数个数字，在大顶堆和小顶堆根节点的和的平均值就是中位数，若两个堆里共有奇数个数字，则小顶堆根节点就是中位数。

注意⚠️ 所以每次塞值，并不是直接塞进去，而是从另一个堆中poll出一个最大（最小）的塞值
两个堆的数目为偶数的时候，将这个值插入大顶堆中，再将大顶堆中根节点（即最大值）插入到小顶堆中；由此保存在小顶堆里面的都是较大的值。
两个堆的数目为奇数的时候，将这个值插入小顶堆中，再讲小顶堆中根节点（即最小值）插入到大顶堆中；由此保存在大顶堆里面的都是较小的值。

import java.util.*;
public class Solution {
    private PriorityQueue<Integer> minHeap =new PriorityQueue<Integer>();//默认是小顶堆
    private PriorityQueue<Integer> maxHeap =new PriorityQueue<Integer>(20,new Comparator<Integer>(){
        public int compare(Integer o1,Integer o2){
            return o2-o1;
        }
    });//改写comparator函数把堆变成大顶堆。
    int count=0;
    public void Insert(Integer num) {
        if(count%2==0){
            maxHeap.offer(num);
            minHeap.offer(maxHeap.poll());
        }
        else{
            minHeap.offer(num);
            maxHeap.offer(minHeap.poll());
        }
        count++;
    }
    public Double GetMedian() {
        if(count%2==0)
            return new Double(minHeap.peek()+maxHeap.peek())/2;
        else
            return new Double(minHeap.peek());
    }
}

其他的题题解
剑指offer题解 java版