LINTCODE——滑动窗口的中位数

本文介绍了一种使用两个堆(multiset)实现的滑动窗口中位数算法。该算法通过维护两个堆(左堆和右堆),使得左堆始终包含窗口内较小的一半元素,右堆包含较大的一半,从而实现在数据流中快速找到中位数的目标。适用于需要实时处理大量数据的场景。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

LINTCODE——滑动窗口的中位数

思路:跟数据流中位数的中位数一样(我之前发的数据流中位数不是用堆来实现的),建立两个堆,left存放数组的前半部分,right存放后半部分,则中位数就为left的最大值;

class Solution {
public:
    /*
     * @param nums: A list of integers
     * @param k: An integer
     * @return: The median of the element inside the window at each moving
     */
    vector<int> medianSlidingWindow(vector<int> &nums, int k) {
        // write your code here
        multiset<int> left,right;
        int n = nums.size();
        vector<int> res;
        if(n == 0)
            return res;

        bool flag = true;
        for(int i = 0 ; i < k ; i++)
        {
            //建立两个堆,左堆存放数组的前半部分,右堆存放后半部分
            build(left,right,flag,nums[i]);

        }    
        res.push_back(*left.rbegin());
        for(int i = k ; i < n ;i++)
        {
            //从堆中删除前一个数;
            erasebuild(left,right,flag,nums[i-k]);
            build(left,right,flag,nums[i]);
            res.push_back(*left.rbegin());

        }
        return res;
    }
    void build(multiset<int> &left, multiset<int> &right, bool &flag, int number)
    {
        if(flag)
        {
            if(!right.empty() && *right.begin() < number)
            {
                right.insert(number);
                number = *right.begin();
                right.erase(right.find(number));
            }
            left.insert(number);

        }
        else
        {
            if( !left.empty() && *left.rbegin() > number)
            {
                left.insert(number);
                number = *left.rbegin();
                left.erase(left.find(number));
            }
            right.insert(number);
        }
        flag = !flag;
    }
    void erasebuild(multiset<int> &left, multiset<int> &right, bool &flag, int number)
    {
        if(left.find(number) != left.end())
        {
            left.erase(left.find(number));
            //如果从左堆删除,则置flag为TRUE,反之为false;
            flag = true;
        }
        else
        {
            right.erase(right.find(number));
            flag = false;
        }

    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值