Skip to content

Latest commit

 

History

History
59 lines (43 loc) · 2.18 KB

File metadata and controls

59 lines (43 loc) · 2.18 KB

题目

给定一个数组 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 ≤ 输入数组的大小,且输入数组不为空。

进阶:

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

解答

使用一个两端队列,队列头部始终保存当前滑动窗口最大值

对于数组中任意元素,从尾部向前比较,如果该元素大于尾部元素,则从队列尾部删除,一直删除完所有比新元素小的元素,然后从尾部插入新元素(的下标)。由于新元素不一定比首元素大,如果首元素刚好从滑动窗口移除时,也要删除首元素,所以在完成上述操作后,也要判断首元素是否已经脱离滑动窗口因为要判断队首元素是否脱离滑动窗口,所以队列存的是元素的下标,同时要从首部删除,所以使用双端队列

当遍历到数组的第k个元素开始,后面每遍历一个元素时,经过上述处理后,就根据队列首部最大值的下标找到最大值,然后添加到结果中

class Solution {
public:
    vector<int> maxSlidingWindow(vector<int>& nums, int k) {
        if(nums.empty() || k <= 0)  return vector<int>();
        
        vector<int> res;
        deque<int> max_deque;

        for(int i = 0;i < nums.size();i++){
            while(!max_deque.empty() && nums[i] > nums[max_deque.back()]) {max_deque.pop_back();}
            max_deque.push_back(i);
            if(i - max_deque.front() + 1 > k)   max_deque.pop_front();
            if(i >= k - 1)
                res.push_back(nums[max_deque.front()]);
        }       
        return res;
    }
};