(C++)剑指offer-63:数据流中的中位数(大小堆维护)

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

大小堆的经典应用,利用大小堆求解一个不断更新的数组的中位数。核心是利用大根堆存储前0到n/2+1个小的数字,小根堆存储剩下大的数字,这样大根堆和小根堆的顶部就是中位数所需(奇数个就是大根堆顶,偶数个就是两堆顶均值)。须维护好两个堆中元素的个数关系,元素的大小关系,即大根堆的所有元素理应比小根堆的所有元素要小,如果违反了此规则,就需要进行元素的对换。
堆的时间复杂度为O(logn);具体代码如下:

class Solution {
public:
    
    priority_queue<int> max_heap;
    priority_queue<int, vector<int>, greater<int>> min_heap; //注意小根堆的定义
    
    void Insert(int num)
    {
        max_heap.push(num);
        if(min_heap.size() && max_heap.top() > min_heap.top()){ //1.大,交换
            auto maxv = max_heap.top(), minv = min_heap.top();
            max_heap.pop(), min_heap.pop();
            max_heap.push(minv), min_heap.push(maxv);
        }
        if(max_heap.size() > min_heap.size() + 1){ //2.多,转移
            min_heap.push(max_heap.top());
            max_heap.pop();
        }
    }

    double GetMedian()
    { 
        if(max_heap.size() + min_heap.size() & 1) return max_heap.top();
        return (max_heap.top() + min_heap.top()) / 2.0;
    }

};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值