AcWing 54 数据流中的中位数

题目描述：

如何得到一个数据流中的中位数？

如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。

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

样例

输入：1, 2, 3, 4

输出：1,1.5,2,2.5

解释：每当数据流读入一个数据，就进行一次判断并输出当前的中位数。

分析：

维护一个大根堆和一个小根堆，大根堆存储较小的那部分数，小根堆存储较大的那部分数。

对于插入操作：先统一插入大根堆，如果大根堆堆顶元素大于小根堆堆顶元素，则交换两个堆顶。当大根堆元素个数比小根堆元素个数多2时，要将大根堆堆顶元素取出插入到小根堆中，以防止两个堆元素个数差大于1.

对于读取中位数操作：若两个堆元素一样多，则输出堆顶的平均值即可，否则输出大根堆堆顶元素。

class Solution {
public:
    priority_queue<int> maxh;
    priority_queue<int,vector<int>,greater<int> > minh;
    void insert(int num){
        maxh.push(num);
        if(minh.size() && minh.top() < maxh.top()){
            int x = minh.top(),y = maxh.top();
            minh.pop(),maxh.pop();
            minh.push(y),maxh.push(x);
        }
        if(maxh.size() == minh.size() + 2){
            int x = maxh.top();
            maxh.pop();
            minh.push(x);
        }
    }

    double getMedian(){
        if(maxh.size() == minh.size())  return (maxh.top() + minh.top()) / 2.0;
        return maxh.top();
    }
};