题目描述
给你一个整数数组 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]
思路
单调队列,参考:代码随想录
需自己构建一个队列,写清楚pop,push,front3个函数即可
1)push函数:队列要始终保持一个从大到小的列表,所以push操作的时候,需要把比当前值小的值,pop出去
2)pop函数:滑动窗口,需要逐次把最左边的数排出去;当时当前队列中只保留一部分从大到小的值【有些数在push操作中被去掉了】;所以当要排出去的数和队首的数一样时,则pop()
3)front函数:返回队首元素即可
from collections import deque
class Myqueue():
def __init__(self):
self.queue = deque()
def push(self, x):
while self.queue and x>self.queue[-1]: # 这里是-1,一点点去掉
self.queue.pop()
self.queue.append(x)
def pop(self, x):
if self.queue and x==self.queue[0]:
self.queue.popleft()
def front(self):
return self.queue[0]
class Solution(object):
def maxSlidingWindow(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
que = Myqueue()
result = []
for i in range(k):
que.push(nums[i])
result.append(que.front())
for j in range(k,len(nums)):
que.pop(nums[j-k]) # 这里是j-k
que.push(nums[j])
result.append(que.front())
return result
if __name__ == '__main__':
s = Solution()
nums = [1, 3, -1, -3, 5, 3, 6, 7]
k = 3
print(s.maxSlidingWindow(nums, k))
记忆点:
需要自己单独建立一个单调队列deque()【这个是个类】
push是始终维护一个从大到小的数组
pop如果不一样,说明在push操作里面早就pop出去了