给定一个数组 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
deque是一个两端开口的队列,用来保存有可能是滑动窗口最大值的数字的下标(index)。在存入一个数字的下标之前,首先要判断队列里已有数字是否小于要存入的数字。如果已有的数字小于待存入的数字,那么这些数字就已经不可能是滑动窗口的最大值,因此它们将会被依次从队列的尾部删除(.pop),同时,如果队列头部的数字已从窗口中滑出,那么滑出的数字也需要从队列的头部删除(.pop(0)),由于队列的头部和尾部都需要删除数字,所以需要双端开口的队列。
在整个过程中,始终保持deque[0]为最大值的index
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
deque=[]
result=[]
for i in range(len(nums)):
# 只存有可能成为最大值数字的index进入deque
while deque and nums[i] > nums[deque[-1]]:
deque.pop()
deque.append(i)
#如果相聚超过窗口长度k,丢弃掉deque中首位index
while i-deque[0]>k-1:
deque.pop(0)
# 只要满足一个窗口的长度k,就向result中添加nums[deque[0]一个结果
if i>=k-1:
result.append(nums[deque[0]]) #在整个过程中,始终保持deque[0]为最大值的index
return result