滑动窗口的最大值

给定一个数组和滑动窗口的大小，请找出所有滑动窗口里的最大值。

例如，如果输入数组 [2,3,4,2,6,2,5,1]及滑动窗口的大小 3，那么一共存在 6 个滑动窗口，它们的最大值分别为 [4,4,6,6,6,5]。

注意：

• 数据保证 k大于 0，且 k 小于等于数组长度。

样例

输入：[2, 3, 4, 2, 6, 2, 5, 1] , k=3

输出: [4, 4, 6, 6, 6, 5]

 解题思路

维护一个双向单调队列，队列放的是元素的下标。我们假设该双端队列的队头是整个队列的最大元素所在下标，至队尾下标代表的元素值依次降低。初始时单调队列为空。随着对数组的遍历过程中，每次插入元素前，首先需要看队头是否还能留在队列中，如果队头下标距离i超过了k，则应该出队。同时需要维护队列的单调性，如果nums[i]大于或等于队尾元素下标所对应的值，则当前队尾再也不可能充当某个滑动窗口的最大值了，故需要队尾出队。始终保持队中元素从队头到队尾单调递减。依次遍历一遍数组，每次队头就是每个滑动窗口的最大值所在下标。

class Solution {
public:
    vector<int> maxInWindows(vector<int>& nums, int k) {
        vector<int> res;
        deque<int> q;
        for(int i=0;i<nums.size();i++){
            if(!q.empty()&&i-q.front()>=k) q.pop_front();
            while(!q.empty()&&nums[q.back()]<nums[i]) q.pop_back();
            q.push_back(i);
            if(i>=k-1){
                res.push_back(nums[q.front()]);
            }
        }
        return res;
        
    }
};

Python代码 

from collections import deque
class Solution(object):
    def maxInWindows(self, nums, k):
        if not nums:
            return []
        n = len(nums)
        res = []
        q = deque()
        for i in range(n):
            if q and i-q[0]+1>k: #遍历的数为窗口大小
                q.popleft()
            while q and nums[q[-1]]<=nums[i]:#判断大小，留下最大的数
                q.pop()
            q.append(i)
            if i-k+1>=0:#结束一个窗口将最大值返回给结果
                res.append(nums[q[0]])
        return res

Java代码

class Solution {
    public int[] maxInWindows(int[] nums, int k) {
        int[] res = new int[nums.length-k+1];
        if(k==0)return res;
        int start;
        int index = 0;
        ArrayDeque<Integer> max = new ArrayDeque<>();
        for(int i=0;i<nums.length;i++){
            start = i-k+1;
            if(max.isEmpty())
                max.add(i);
            else if(start>max.peekFirst())
                max.pollFirst();

            while((!max.isEmpty())&&nums[max.peekLast()]<=nums[i])
                max.pollLast();
            max.add(i);
            if(start>=0)
                res[index++] = nums[max.peekFirst()];
        }
        return res;
    }
}