题目描述
给你一个整数数组 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))