中位数操作

本题源自leetcode

---------------------------------------------------

思路 用俩个优先队列存储。 优先队列 是按降序排列的

1 当 加入一个数字时。先把数字加入小的队列中。然后弹出小队列的队头,并把其相反数压入大的队列。如果小队列的个数小于大队列。就把大的队列头弹栈,取反后压入小的。

2 这样就保证了,如果小的个数大于大的个数,则为奇数,切中位数在小队列的队头。否则就是偶数。中位数是小堆头加上取反的大队头求平均数


代码:

class MedianFinder {
    priority_queue small,large;
public:
    /** initialize your data structure here. */
    MedianFinder() {
        
    }
    
   void addNum(int num) {
        small.push(num);
        large.push(-small.top());
        small.pop();
        if (small.size() < large.size()) {
            small.push(-large.top());
            large.pop();
        }
    }

    double findMedian() {
        return small.size() > large.size()
               ? small.top()
               : (small.top() - large.top()) / 2.0;
    }
};


你可能感兴趣的:(C++,leetcode,算法)