劍指Offer-59 滑動視窗的最大值
阿新 • • 發佈:2018-12-10
題目:
給定一個數組和滑動視窗的大小,請找出所有滑動窗口裡的最大值。
例如,如果輸入陣列[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