剑指offer-数据流中的中位数

问题

题目：[剑指offer-数据流中的中位数]

思路

这个题目的意思是，数据流中一直有数据在插入，怎么样实时返回中位数。这个题目还是比较有实际意义，因为像在线系统一般都是实时插入数据。
这个题目是借助辅助数据结构，堆去完成。
对于一个数据，借助两个堆来进行排序，并且两者的元素之差始终不超过1个。
这个题目还是比较有技巧的！
bug也改了我好长时间，就是要小心max_heap上来为空的情形，因此此时不用判断min_heap是否为空。它肯定为空。
每个堆插入元素的时候都需要判断是否破坏了整个数组的有序性

代码

#include <iostream>
#include <queue>
using namespace std;
class Solution {
private:
    priority_queue< int, vector<int>, less<int> > max_heap; // less element
    priority_queue< int, vector<int>, greater<int> > min_heap; // greater element
public:
    void Insert(int num)
    {
        int sz = max_heap.size() + min_heap.size();
        if( sz%2 ){
            // add num to min_heap

            if( num > max_heap.top() ) min_heap.push(num);
            else{
                max_heap.push(num);
                num = max_heap.top();
                max_heap.pop();

                min_heap.push( num );
            }
        }else{
            // add num to max_heap
            if( max_heap.empty() ) { 
                max_heap.push(num); 
                return; 
            }
            if( num < min_heap.top() ) max_heap.push( num );
            else{
                min_heap.push( num );
                num = min_heap.top();
                min_heap.pop();

                max_heap.push( num );
            }
        }
    }

    double GetMedian()
    { 
        int sz1 = max_heap.size();
        int sz2 = min_heap.size();
        if( sz1 == sz2 ) return 1.0/2*( max_heap.top() + min_heap.top() );
        else if( sz1 + 1 == sz2 ){
            return min_heap.top();
        }
        else return max_heap.top();
    }
};