295. 数据流的中位数

中位数是有序整数列表中的中间值。如果列表的大小是偶数，则没有中间值，中位数是两个中间值的平均值。

例如 arr = [2,3,4] 的中位数是 3 。
例如 arr = [2,3] 的中位数是 (2 + 3) / 2 = 2.5 。
实现 MedianFinder 类:

MedianFinder() 初始化 MedianFinder 对象。

void addNum(int num) 将数据流中的整数 num 添加到数据结构中。

double findMedian() 返回到目前为止所有元素的中位数。与实际答案相差 10-5 以内的答案将被接受。

示例 1：

输入
["MedianFinder", "addNum", "addNum", "findMedian", "addNum", "findMedian"]
[[], [1], [2], [], [3], []]
输出
[null, null, null, 1.5, null, 2.0]

解释
MedianFinder medianFinder = new MedianFinder();
medianFinder.addNum(1);    // arr = [1]
medianFinder.addNum(2);    // arr = [1, 2]
medianFinder.findMedian(); // 返回 1.5 ((1 + 2) / 2)
medianFinder.addNum(3);    // arr[1, 2, 3]
medianFinder.findMedian(); // return 2.0

代码

class MedianFinder {
private:
    //大顶堆
    priority_queue<int> que1;

    priority_queue<int,vector<int>,greater<int>> que2;
    
    int capacity;
public:
    //寻找中间索引
    MedianFinder() {
        capacity = 0;
    }
    
    void addNum(int num) {
        if(!que2.empty() && num >= que2.top()){
            que2.push(num);
        }else{
            que1.push(num);
        }
        if(que1.size() > que2.size()){
            int l = que1.top();que1.pop();
            que2.push(l);
        }else if(que1.size() < que2.size()){
            int r = que2.top();que2.pop();
            que1.push(r);
        }
        capacity++;
    }
    
    double findMedian() {
        int size1 = que1.size();
        int size2 = que2.size();
        if(capacity % 2 == 0){
            double l = que1.top();
            double r = que2.top();
            return (double)((l + r)/2);
        }else{
            if(size1 > size2){
                return que1.top();
            }else{
                return que2.top();
            }   
        }
    }
};