【leetcode - 59-I】滑动窗口的最大值 -Easy

给定一个数组 nums 和滑动窗口的大小 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

提示:

你可以假设 k 总是有效的,在输入数组不为空的情况下,1 ≤ k ≤ 输入数组的大小。

【original】

思路:暴力解法,遍历得到每个窗口的最大值

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        stack = []
        if not nums:
            return []
        def maxx(x):
            a = x[0]
            for i in x:
                if i>=a:
                    a = i
            return a
        for i in range(0,len(nums)-k+1):
            stack.append(maxx(nums[i:i+k]))
        return stack
original

【upgrade】

思路: 分几种情况,如果滑动窗口滑到当前窗口,出栈的那个元素为上一个的最大值,则计算当前最大值,如果出栈元素不是最大值,判断新入元素和上一个max的大小作为最大值,如果大于新入元素则仍为之前的值,小于则为新入的值

class Solution:
    def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
        if not nums:
            return []
        res = [max(nums[:k])]
        i = 1
        j = k+1
        while j<=len(nums):
            if nums[j-1]>res[-1]:
                res.append(nums[j-1])
            elif nums[i-1]==res[-1]:
                res.append(max(nums[i:j]))
            else:
                res.append(res[-1])
            i+=1
            j+=1
        return res
Upgrade

猜你喜欢

转载自www.cnblogs.com/akassy/p/12632600.html