LeetCode 239. 滑动窗口最大值

题目描述

给定一个数组 nums，有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口 k 内的数字。滑动窗口每次只向右移动一位。

返回滑动窗口最大值。

示例:

输入: nums = [1,3,-1,-3,5,3,6,7], 和 k = 3
输出: [3,3,5,5,6,7] 
解释: 

  滑动窗口的位置                最大值
---------------               -----
[1  3  -1] -3  5  3  6  7       3
 1 [3  -1  -3] 5  3  6  7       3
 1  3 [-1  -3  5] 3  6  7       5
 1  3  -1 [-3  5  3] 6  7       5
 1  3  -1  -3 [5  3  6] 7       6
 1  3  -1  -3  5 [3  6  7]      7

注意：

你可以假设 k 总是有效的，1 ≤ k ≤ 输入数组的大小，且输入数组不为空。

进阶：

你能在线性时间复杂度内解决此题吗？

问题分析

创建一个双端队列deque。然后遍历给定数组nums，对于当前遇到的数组元素，先判断如果deque不为空并且队首元素已经不在窗口范围内了，那么将deque的队首元素删除。接下来将当前数组元素nums[i]逐个与队列中从deque的尾部开始的元素比较，如果大于等于，那么将deque中的此元素删除，循环进行直到在deque中遇到比nums[i]大的元素或者deque空了为止，所以deque一直都是从队首单调递减的。接下来把 i 加入队列deque的尾部，那么此时deque的队首元素的值就是当前窗口的最大值，将其加入到答案数组ans的后面。当给定数组nums遍历结束后，返回ans。

代码实现

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