剑指offer 数据流中的中位数

这个题还是很不错的，是对大小堆的一个灵活运用

题目描述
如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。

做法就是用一个大顶堆和一个小顶堆，维持大顶堆的数都小于等于小顶堆的数，且两者的个数相等或差1。平均数就在两个堆顶的数之中。

java代码

PriorityQueue<Integer> small = new PriorityQueue<>();
    PriorityQueue<Integer> big = new PriorityQueue<>(new Comparator<Integer>() {
        @Override
        public int compare(Integer o1, Integer o2) {
            return o2 - o1;
        }
    });
    int count = 0;
    //1 2 3 4 5 6 7 8
    public void Insert(Integer num) {

        if(count % 2 == 0) {
            big.offer(num);
            int tmp = big.poll();
            small.offer(tmp);
        }else {
            small.add(num);
            int tmp = small.poll();
            big.offer(tmp);
        }
        count++;
    }

    public Double GetMedian() {
        if(count == 0) return 0d;
        if(count % 2 == 0) return new Double((big.peek()+small.peek())) / 2;
        else return new Double(small.peek());
    }