滑动窗口的最大值

滑动窗口的最大值

题目描述

给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{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]}。

解题思路

• 使用双端队列来维护滑动窗口，队首存放当前滑动窗口的最大值的下标
• 每一个新的元素需要与头尾元素比较
• 与队尾元素比较，将所有比新来元素小的剔除，因为它们不可能是滑动窗口的最大值
• 与队首元素比较，将所有过期的元素剔除，即当队首元素与当前元素的距离超过滑动窗口大小时，将其剔除

class Solution {
public:
    vector<int> maxInWindows(const vector<int>& num, unsigned int size)
    {
        vector<int> ans;
        deque<int>dq;    //双端队列的队首存放当前滑动窗口最大值的下标
        if(num.empty() || size<1) return ans;
        for(int i=0;i<num.size();i++){
            //双端队列的末尾小于当前值，所以不可能是最大值，剔除
            while(dq.size() && num[dq.back()] < num[i]) 
                dq.pop_back();
            //当前最大值过期，就要剔除，dq.front存放的是下标
            while(dq.size() && i-dq.front()+1 >size) 
                dq.pop_front();
            dq.push_back(i);
            //当滑动窗口大于等于size时，才开始写入最大窗口值
            if( size && i+1 >=size) 
                ans.push_back(num[dq.front()]); //双端队列队首存放最大值的下标
        }
        return ans;
    }
};