剑指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

猜你喜欢

转载自blog.csdn.net/u013796880/article/details/84929149