【剑指offer中等部分26】数据流中得中位数（java）

题目描述

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

分析

这题挺难的，我是看了别人得文章，理解算法后，再编了一遍。
关于大小顶堆概念见：https://www.cnblogs.com/lanhaicode/p/10546257.html

主要思路：
主要的思想是：因为要求的是中位数，那么需要两个堆，大顶堆用来存较小的数，小顶堆存较大的数，并且保证过程中大小顶堆中数据数目最多相差1，代码如下：
（仔细阅读代码与注释）

import java.util.TreeSet;
public class Solution {
    //注意此处要求小顶堆中最小的，都要大于等于大顶堆中最大的，如小顶堆[6,8,9]，大顶堆[1,3,4]
    //保证大小顶堆中数字个数最多只能相差1
    TreeSet<Integer> max = new TreeSet<Integer>();  //大顶堆，用于存放前面一半的元素(小的)
    TreeSet<Integer> min = new TreeSet<Integer>();  //小顶堆，用于存放后面一般的元素(大的)
    public void Insert(Integer num) {
        if (((max.size() + min.size()) & 1) == 0) { //两堆相加为偶数，即10，110，1110等，反正第一位不是0，所以&后得0
            //num值大于小顶堆中最小的值，则小顶堆根节点值(最小值)放入小顶堆，如num=7,7>6,则将7加入小顶堆，再把6加入大顶堆
            if (min.size() > 0 && num > min.first()) {
                min.add(num); //值插入小顶堆中
                num = min.first();
                min.remove(num);//拿出小顶堆中的根节点值，
            }
            max.add(num);
        }else {
            //如小顶堆[7,8,9]，大顶堆[1,3,4,6]，num=5, 5<6，则将6加入小顶堆，变为[6,7,8,9]，[1,3,4,5]
            if (max.size() > 0 && num < max.last()) {
                max.add(num);
                num = max.last();
                max.remove(num);
            }
            min.add(num);
        }
    }

    public Double GetMedian() {
        int size = max.size() + min.size();
        if (size == 0) {
            return 0.0;
        }
        if ((size & 1) == 0) {
            return (max.last() + min.first()) / 2.0; //偶数求和，如[6,7,8,9]，[1,3,4,5]情况，取(5+6)/2.0
        }else {
            return (double)max.last();//奇数,如小顶堆[7,8,9]，大顶堆[1,3,4,6]情况，取6.0
        }
    }
}