滑动窗口的最大值(python)

剑指 Offer 59 - I. 滑动窗口的最大值

给定一个数组 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

def func(nums, k):
    if len(nums) == 0 or len(nums) == 1:
        return nums
    elif len(nums) <= k:
        return [max(nums)]
    n = len(nums)
    dp = []
    for i in range(n - k + 1):
        dp.append(max(nums[i:i + k]))
    return dp

print(func(nums = [1,3,-1,-3,5,3,6,7], k = 3))