思路1:暴力法
时间复杂度:O(Nk)。其中 N 为数组中元素个数。
空间复杂度:O(N),用于输出数组和滑动窗口。
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
ans = []
window = []
for i in range(len(nums)):
if i<k-1:
window.append(nums[i])
else:
window.append(nums[i]) # k个
ans.append(max(window))
window.pop(0) # k-1个
return ans
思路2:基于双端队列实现一个单调队列,可以快速插入、删除和求最大值。
时间复杂度:O(N)。其中 N 为数组中元素个数。
空间复杂度:O(N),用于输出数组和滑动窗口。
from collections import deque
# 根据下面的需要,主要需要实现下面三个函数
class Queue(deque):
# 使用双端队列构造一个单调的递减队列
def pushItem(self,num):
while self and num>self[-1]: # 虽然这里有循环,但是整体还是O(n)
self.pop()
self.append(num)
def popItem(self,num):
if self[0]==num:
self.popleft()
def max(self):
return self[0]
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
ans = []
window = Queue()
for i in range(len(nums)):
if i < k - 1:
window.pushItem(nums[i])
else:
window.pushItem(nums[i])
ans.append(window.max())
window.popItem(nums[i - k + 1])
return ans