滑动窗口的最大值(max pooling的实现)

  • 题目:给定一个数组和滑动窗口的大小,找出所有滑动窗口里数值的最大值。例如,如果输入数组{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]}。
  • idea:
  1. 采用一个双向队列来保存当前窗口的可能最大值的下标
  2. 滑动到新的数字时,从队列尾开始遍历,假如队列中的下标所在数字比当前数字小,说明这个数字不可能成为最大值了,因此应该删除。直到在队列中找到一个数字比当前值大(队列里边这个数字之前的数字都比这个数字大)
  3. 同时还要判断队首(只需要判断一次,因为滑动stride=1,每次最多有一个下标滑出)是否已经滑出窗口
  • code
#include <bits/stdc++.h>
class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> ans;
        deque<int> idx;
        int len = num.size();
        if (len<size || !len || size<1) return ans;
        for (int i = 0; i < len; i++){
            while (!idx.empty() && num[i]>=num[idx.back()]) idx.pop_back();
            if (!idx.empty() && idx.front()<=i-(int)size) idx.pop_front();
            idx.push_back(i);
            if (i>(int)size-2) ans.push_back(num[idx.front()]);
        }
        return ans;
    }
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值