leetcode 295 数据流的中位数（堆排序）

题目描述：
中位数是有序列表中间的数。如果列表长度是偶数，中位数则是中间两个数的平均值。

例如，

[2,3,4] 的中位数是 3

[2,3] 的中位数是 (2 + 3) / 2 = 2.5

设计一个支持以下两种操作的数据结构：

void addNum(int num) - 从数据流中添加一个整数到数据结构中。
double findMedian() - 返回目前所有元素的中位数。

思路：
利用两个堆排序，一个大顶堆，一个小顶堆
大顶堆特点是里面的元素都是按照从大到小的顺序排列
小顶堆特点是里面的元素都是按照从小到大的顺序排列

例一：3124
大顶堆：21
小顶堆：34
返回（2+3)/2

例一：31542
大顶堆：321
小顶堆：45
返回 3

代码如下：

class MedianFinder {
public:
    priority_queue<int>max_num;
    priority_queue<int,vector<int>,greater<int>>min_num;
    /** initialize your data structure here. */
    MedianFinder() {

    }
    
    void addNum(int num) {
        max_num.push(num);
        min_num.push(max_num.top());
        max_num.pop();
        if(max_num.size()<min_num.size()){
            max_num.push(min_num.top());
            min_num.pop();
        }
    }
    double findMedian() {
        if(max_num.size()==min_num.size()){
            return (max_num.top()+min_num.top())/2.0;
        }
        else return max_num.top();
    }
};