【剑指offer】数据流中的中位数（数组）

题目描述

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

思路

用一个最大堆和一个最小堆实现。插入效率为O(logN)，查找效率为O(1)。
最大堆的元素数不少于最小堆，并且相差不大于1。

用STL中的priority_queue实现堆

priority_queue<int, vector<int>, greater<int>> min; //最小堆
priority_queue<int, vector<int>, less<int>> max; //最大堆

代码

class Solution {
private:
    priority_queue<int, vector<int>, greater<int>> min;
    priority_queue<int, vector<int>, less<int>> max;
public:
    void Insert(int num){
        if(max.empty() || num<= max.top())
            max.push(num);
        else
            min.push(num);
        // 保持最大堆数目不小于最小堆；并且相差最多为1
        if(max.size()== min.size() +2){
            min.push(max.top());
            max.pop();
        }
        if(max.size() +1 == min.size()){
            max.push(min.top());
            min.pop();
        }
    }

    double GetMedian(){
        if(min.size() == max.size())
            return (min.top() + max.top()) / 2.0;
        else
            return max.top();
    }
};