题目:
给定一个数组和滑动窗口的大小,请找出所有滑动窗口里的最大值。
例如,如果输入数组[2, 3, 4, 2, 6, 2, 5, 1]及滑动窗口的大小3,那么一共存在6个滑动窗口,它们的最大值分别为[4, 4, 6, 6, 6, 5]。
注意:
数据保证k大于0,且k小于等于数组长度。
样例
输入:[2, 3, 4, 2, 6, 2, 5, 1] , k=3
输出: [4, 4, 6, 6, 6, 5]
解答:
很奇怪,这道题在牛客网和acwing上提交总是错误,但是在leetcode上能够成功提交(https://leetcode.com/problems/sliding-window-maximum/)。
import collections
class Solution(object):
def maxSlidingWindow(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
if k == 1:
return nums
d = []
q = collections.deque()
for i, v in enumerate(nums):
if not q:
q.append(i)
else:
if nums[q[0]] <= v:
while(q):
q.pop()
q.append(i)
else:
while(nums[q[-1]] < v):
q.pop()
q.append(i)
if i >= k - 1:
d.append(nums[q[0]])
if q[0] < i + 2 - k:
q.popleft()
return d