1. 题目
[原题链接]给定一个数组 nums 和滑动窗口的大小 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 ≤ 输入数组的大小。
注意:本题与LeetCode主站 239 题相同:https://leetcode-cn.com/problems/sliding-window-maximum/
2. 思路
思路1:
采用蛮力法,扫描每个滑动窗口的所有数字并找出其中的最大值。如果滑动窗口的大小为k,则需要 O(k)时间;如果输入数组的长度为n,这种算法的总时间复杂度是O(nk)。
思路2:
一个滑动窗口可以看成一个队列。当窗口滑动时,窗口的第一个数字被删除,同时窗口的末尾添加一个新的数字,这符合队列FIFO的特性。JZ09讨论了如何用两个栈实现一个队列,JZ30实现了用O(1)时间得到最小值的栈,综合这两个问题得到的算法,总时间复杂度是O(n)。
思路3:
类比JZ30中利用一个单调的辅助栈实现了 O(1)时间内获取栈内最小值。同理,本题可以利用一个单调的辅助队列获取队列(即滑动窗口)内的最大值。下表是一个具体的例子。
我们希望辅助队列的队首元素始终是滑动窗口最大值,这意味着在遍历数组时,辅助队列要保持①严格递减且②仅包含窗口内的元素。当一个数字不再可能是滑动窗口的最大值时,要把它从队尾删除;当一个数字已经不处于滑动窗口中了,要把它从队首删除。由于队首和队尾都有可能删除数字,所以应选择双端队列(deque)作为辅助队列。另外,为了知道数字是否还处于滑动窗口中,应在deque中存入数字在数组中的下标,而非数值。
3. 实现
思路1:
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
res = []
for i in range(len(nums)):
if i + k <= len(nums):
res.append(max(nums[i:i+k]))
else:
break
return res
思路3:
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
deque,res = [],[] # deque也可以用collection里的双端队列实现
for i in range(0, len(nums)):
while deque and nums[i]>nums[deque[-1]]: # 只存有可能成为最大值的数字的index进deque
deque.pop()
deque.append(i)
while i-deque[0]>k-1: # 是否还处于滑动窗口中
deque.pop(0)
if i >= k-1:
res.append(nums[deque[0]])
return res
4. 其它
- 题目难度:简单
- 相关题目: