1. 程式人生 > >劍指Offer-59 滑動視窗的最大值

劍指Offer-59 滑動視窗的最大值

題目:

給定一個數組和滑動視窗的大小,請找出所有滑動窗口裡的最大值。
例如,如果輸入陣列[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