一、题目
给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 k 个数字。滑动窗口每次只向右移动一位。
返回 滑动窗口中的最大值 。
示例 1:
输入: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
示例 2:
输入:nums = [1], k = 1
输出:[1]
提示:
1 <= nums.length <= 10 ^5
-10 ^4 <= nums[i] <= 10 ^4
1 <= k <= nums.length
二、解法
如何保证元素都是窗口内的?记录已经访问的元素下标,然后遍历其他元素,利用窗口大小k来判断
使用队列,队列中队头元素记录当前窗口的最大值对应的下标。
保持队列中的都是当前窗口内的元素,而没有以前的元素。
时间复杂度O(n)
空间复杂度O(n)
完整代码
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
res = []
q = deque()
for i, x in enumerate(nums):
while q and nums[q[-1]] < x:
q.pop()
q.append(i)
while q and q[0] <= i - k:
q.popleft()
if i >= k - 1:
res.append(nums[q[0]])
return res