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

思路

• 数组为nums，窗口大小为k。使用一个双向队列，用来存放数的index，保证队列头部为当前窗口中最大值的index。
• 先初始化窗口，即将前k个数放入放入队列中。规则是，一个数n从尾部插入队列，如果n大于队列尾部的数，那么尾部的数出队列。直到队列为空，或者队列尾部的数大于n，则n从尾部进入队列。
• 窗口移动的时候，先检查队列首部的index是否在窗口中，不在的话出队列。然后从尾部插入一个值，按照上一步。

代码

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        if ( !nums.size() || k > nums.size() )
            return {};

        vector<int> ans;
        deque<int> dq;

        int left = 0, right = 0;

        while ( right < nums.size() ) {
            if ( right < k ) {
                while ( !dq.empty() && nums[right] > nums[dq.back()] )
                    dq.pop_back();
                
                dq.push_back( right );
                
                if ( right == k - 1 ) {
                    ++left;
                    ans.push_back( nums[dq.front()] );
                }
            }
            else {
                if ( dq.front() < left )
                    dq.pop_front();
                
                while ( !dq.empty() && nums[right] > nums[dq.back()] )
                    dq.pop_back();
                
                dq.push_back( right );
                ans.push_back( nums[dq.front()] );

                ++left;
            }

            ++right;
        }

        return ans;
    }
};