直观暴力解法:维护一个长度k的滑动窗口,遍历nums(长度为n),每次找窗口里的最大值,总复杂度为O(nk),k过大可能超时
这题用单调队列——队列里元素是单调的。
维护一个单调递减的双端队列deque,队列头永远是当前的最大值
如此理解:队列里只存放当前窗口内,可能成为最大值的那些元素。
如此操作:
1、当一个元素num要加入队列时,从队列的右侧开始考察,把比num小的都pop了,因为num比他们更大,只要num加入进了队列,他们永远不可能成为最大值了
2、前期经典滑动窗口模板:
1、加入right(这里有条件)
2、达到条件开始收割答案、并且收缩窗口(注意,收缩时要看扔掉的是不是当前队头)
3、right += 1,滑动
class Solution:
from collections import deque
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
res = []
my_que = deque()
left = 0
right = 0
while right < len(nums):
#加入
while my_que and my_que[-1] < nums[right]:
my_que.pop()
my_que.append(nums[right])
#达到条件,收割,收缩
if right - left + 1 == k:
res.append(my_que[0])
#收缩时要看,队头要不要pop掉
if nums[left] == my_que[0]:
my_que.popleft()
left += 1
#滑动
right += 1
return res