面试题64. 数据流中的中位数

题目描述

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

思路1：
用Partition算法，找到中位数，详细见这篇博文

思路2：
采用插入排序的方法，每读取一个数字时，用插入排序让数组保持排序状态，需要O(n2)
但是在求中位数时，只需要O(1)。Java中可以借用LinkedList实现插入排序

import java.util.LinkedList;

public class Solution {

    private LinkedList<Integer> list = new LinkedList<>();

    /**
     * 将数据插入到LinkedList中
     */
    public void Insert(Integer num) {
        if(list.isEmpty() || num < list.getFirst()) {
            list.addFirst(num);
        }else {
            boolean insertSuccess = false;
            for(Integer i : list) {
                if(num < i) {
                    int index = list.indexOf(i);
                    list.add(index, num);
                    insertSuccess = true;
                    break;
                }
            }
            if(!insertSuccess) {
                list.addLast(num);
            }
        }
    }

    /*
     * 获取中位数
     */
    public Double GetMedian() {
        if(list.isEmpty()) {
            return null;
        }

        int mid = list.size() / 2;

        // 如果长度是偶数
        if((list.size() & 1) == 0) {            
            Integer first = list.get(mid - 1);
            Integer second = list.get(mid);
            return Double.valueOf(first + second) / 2.0;
        }

        return Double.valueOf(list.get(mid));      
    }

}