239. 滑动窗口最大值
原始题目链接:https://leetcode.cn/problems/sliding-window-maximum/
给你一个整数数组 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]
解题思路:
使用单调队列来解决滑动窗口的问题。求最大值,为了方便,保持队列的单调递减的特性,这样队头就是最大值。滑动窗口满了就移出最开始入队的元素,即队头元素。新元素入队,如果新元素比队里的元素(是当前元素前面的元素,即遍历过的元素且在队列中)都大,那么队列中的元素不可能成为最大值,弹出队列中的元素即可。
代码实现:
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
if not nums or k == 0:
return []
if len(nums) == 1:
return [nums[0]]
# 维护一个单调递减的队列,保存的是滑动窗口的值
# 队头元素是最大值,队列里存储的是元素在数组中的下标
queue = collections.deque()
ans = []
# 遍历nums
for i in range(len(nums)):
# 滑动窗口中的元素满了,即元素大小等于k个
# 滑动窗口有k个元素:队头元素的下标queue[0] 和当i-k相等
# 为了能让下个元素进入滑动窗口,需要弹出队头元素,队头元素是最先进入队列的元素
if queue and queue[0] == i - k:
queue.popleft()
# 保持单调递减的特性
# 如果新进入的元素比队列里的元素(必须是当前元素前面的元素)大,那么队列中的元素不可能是接下来滑动窗口的最大值
# 可以将小于当前元素的值出队,再将新元素入队,为了保持递减的特性,从最小的元素开始比较,
# 所以出队的顺序是从右边出,因为最右边的元素最小
while queue and nums[queue[-1]] < nums[i]:
queue.pop()
# 新元素入队
queue.append(i)
# 获取滑动窗口的最大值
# 当前元素的下标和k相差大于等于1,表示滑动窗口已经有k个值
if i >= k - 1:
ans.append(nums[queue[0]])
return ans
参考文献:
https://leetcode.cn/problems/sliding-window-maximum/solution/acm-xuan-shou-tu-jie-leetcode-hua-dong-c-i3wj/