小白如何如何理解滑动窗口最大值问题python

文章目录

题目描述

给你一个整数数组 nums,有一个大小为 k 的滑动窗口从数组的最左侧移动到数组的最右侧。你只可以看到在滑动窗口内的 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

思路

很容易想到暴力求解,遍历数据[i:i+k]取出max最大值即可,但是这种做法容易暴力超时,因此引出了自DIY的队列,主要关注的是什么时候弹出,什么时候加入队列

什么时候弹出元素

1.当遍历个数大于k的时候,弹出queue左侧==nums[i-k]

2.当队列最右侧元素小于nums[i],弹出右侧元素,因为他不可能是最大值了,队列内最多只有k个值,所以遍历到的这个元素才能成为最大值

什么时候加入元素

没遍历一次长度大于k的时候就加元素

代码示例和解释

python 复制代码
from collections import deque
def max SlidingWindow(nums, k):
	queue = deque() # 双端都能插入删除的队列
	res = [] # 存放最后的结果
	# 把前面三个元素按需求加入到deque
	for i in range(k):
		while queue and nums[i] > queue[-1]:
			queue.pop() # 弹出右边的元素,不可能成为最大值
		deque.append(nums[i])
	# 从k开始遍历后面的每个元素
	for i in range(k, len(nums)):
		# 只有遍历到的元素queue[0]等于nums[i-k]才需要弹出元素
		if queue and queue[0] == nums[i-k]:
			queue.popleft()
		while queue and queue[-1] < nums[i]:
			queue.pop()
		queue.append(nums[i])
		res.append(queue[0]) # 加入最大值
	return res

*[注意点]:queue内部的元素个数一定是小于等于k的,比如k = 3的时候,因为遍历到长度第4个元素,如果queue首元素和nums[i-k]元素相等就会有pop操作,即使不相等,其他需要pop()的元素也会被push的过程弹出,所以queue的长度始终不可能大于k

相关推荐
pianmian110 分钟前
完全平方数
数据结构·算法
A_Tai233333312 分钟前
贪心算法解决用最少数量的箭引爆气球问题
算法·贪心算法
唐叔在学习24 分钟前
【唐叔学算法】第19天:交换排序-冒泡排序与快速排序的深度解析及Java实现
java·算法·排序算法
_nirvana_w_24 分钟前
C语言实现常用排序算法
c语言·算法·排序算法
XWXnb628 分钟前
数据结构:栈
数据结构
唐叔在学习31 分钟前
【唐叔学算法】第18天:解密选择排序的双重魅力-直接选择排序与堆排序的Java实现及性能剖析
数据结构·算法·排序算法
Kenneth風车1 小时前
【机器学习(九)】分类和回归任务-多层感知机(Multilayer Perceptron,MLP)算法-Sentosa_DSML社区版 (1)11
算法·机器学习·分类
最后一个bug1 小时前
rt-linux中使用mlockall与free的差异
linux·c语言·arm开发·单片机·嵌入式硬件·算法
木向2 小时前
leetcode22:括号问题
开发语言·c++·leetcode
蹉跎x2 小时前
力扣1358. 包含所有三种字符的子字符串数目
数据结构·算法·leetcode·职场和发展