滑动窗口的最大值(单调队列 / 剑指offer)

原题来自牛客网剑指offer系列:链接

题目描述

给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3,那么一共存在6个滑动窗口,他们的最大值分别为{4,4,6,6,6,5}; 针对数组{2,3,4,2,6,2,5,1}的滑动窗口有以下6个: {[2,3,4],2,6,2,5,1}, {2,[3,4,2],6,2,5,1}, {2,3,[4,2,6],2,5,1}, {2,3,4,[2,6,2],5,1}, {2,3,4,2,[6,2,5],1}, {2,3,4,2,6,[2,5,1]}。

思路

可以直接使用蛮力法,遍历每一种情况的窗口,都找出窗口的最大值,时间复杂度O(n^size)。

也可以采用维护双端单调队列的方法,时间复杂度O(n)。对新来的元素k,将其与双端队列中的元素相比较:
(1)后面比k小的,直接移出队列(因为不再可能成为后面滑动窗口的最大值了!)
(2)前面比k大的X,比较两者下标,判断X是否已不在窗口之内,不在了,直接移出队列

该单调队列保存下标,其中头元素为最大值的下标,且队列中不含不在窗口内的下标。

最终队列的第一个元素是滑动窗口中的最大值。

代码

(1)蛮力法:O(n*size)

class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> ans;
        if(size==0)
            return ans;
        for(int i=0;i<=(int)num.size()-(int)size;i++){
            int max=num[i];
            for(int j=i;j<i+size;j++){
                if(num[j]>max)
                    max=num[j];
            }
            ans.push_back(max);
        }
        return ans;
    }
};

(2)单调队列:O(n)

class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> res;
        deque<int> s;
        for(unsigned int i=0;i<num.size();++i){
            while(s.size() && num[s.back()]<=num[i])//从后面依次弹出队列中比当前num值小的元素,同时也能保证队列首元素为当前窗口最大值下标
                s.pop_back();
            while(s.size() && i-s.front()+1>size)//当当前窗口移出队首元素所在的位置,即队首元素坐标对应的num不在窗口中,需要弹出
                s.pop_front();
            s.push_back(i);//把每次滑动的num下标加入队列
            if(size&&i+1>=size)//当滑动窗口首地址i大于等于size时才开始写入窗口最大值
                res.push_back(num[s.front()]);
        }
        return res;
    }
};

2021.3.4:


class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> res;
        deque<int> s;
        int len=num.size();
        if(size>len)
            return res;
        for(int i=0;i<len;i++){
            while(s.size()&&num[s.back()]<=num[i])
                s.pop_back();
            while(s.size()&&s.front()+size<=i)
                s.pop_front();
            s.push_back(i);
            if(i>=size-1)
                res.push_back(num[s.front()]);
        }
        return res;
    }
};
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值