题目描述
给你一个整数数组 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 31 [3 -1 -3] 5 3 6 7 31 3 [-1 -3 5] 3 6 7 51 3 -1 [-3 5 3] 6 7 51 3 -1 -3 [5 3 6] 7 61 3 -1 -3 5 [3 6 7] 7
示例二
输入:nums = [1], k = 1
输出:[1]
代码实现
# coding:utf-8
# 滑动窗口最大值
# https://leetcode.cn/problems/sliding-window-maximum/class Solution(object):def maxSlidingWindow(self, nums, k):""":type nums: List[int]:type k: int:rtype: List[int]"""ans = []que = []for i in range(len(nums)):while que and ((i - k + 1) > k):del que[0]while que and (nums[que[-1]] < nums[i]):que.pop()que.append(i)if k and (i >= k - 1):ans.append(nums[que[0]])return ansif __name__ == '__main__':s = list(map(int, (input("input arry:").split(','))))k = int(input("input K:").split(' ')[0])solution = Solution()print(solution.maxSlidingWindow(s, k))