题目:239. 滑动窗口最大值
题目描述:
给定一个数组 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
提示:
1. 1 <= nums.length <= 10^5
2. -10^4 <= nums[i] <= 10^4
3. 1 <= k <= nums.length
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/gu-piao-de-zui-da-li-run-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
1、滑动窗口时,每滑动一次,会新加入一个数,同时丢掉一个数。因此需要解决三个数的大小问题,即:当前窗口的最大值、窗口往右滑动时新加入的值和丢掉的值。
2、用一个变量temp记录当前窗口的最大值,先比较新加入的数会不会改变当前窗口的最大值temp,若新加入的数大于temp,更新temp,窗口右滑;新加入的数小于temp,需要比较丢弃的数与temp的大小,若即将丢弃的数等于temp,意味着这个即将被丢弃的数就是当前窗口的最大值,需要先滑动窗口然后重新找到窗口最大值,如果即将丢弃的数小于当前窗口最大值,temp不变,窗口右滑即可。
3、我这里循环退出时还有最后一个窗口的最大值没有加到ans列表中,故退出循环需要加上temp。
代码:
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
n = len(nums)
if n == 0:
return []
ans = []
temp = max(nums[:k])
for i in range(n-k):
ans.append(temp)
if nums[i+k] > temp:
temp = nums[i+k]
else:
if nums[i] == temp:
temp = max(nums[i+1:i+1+k])
ans.append(temp)
return ans