难度: Hard
原题连接
内容描述
Given an array nums, there is a sliding window of size k which is moving from the very left of the array to the very right. You can only see the k numbers in the window. Each time the sliding window moves right by one position. Return the max sliding window.
Example:
Input: nums = [1,3,-1,-3,5,3,6,7], and k = 3
Output: [3,3,5,5,6,7]
Explanation:
Window position Max
--------------- -----
[1 3 -1] -3 5 3 6 7 3
1 [3 -1 -3] 5 3 6 7 3
1 3 [-1 -3 5] 3 6 7 5
1 3 -1 [-3 5 3] 6 7 5
1 3 -1 -3 [5 3 6] 7 6
1 3 -1 -3 5 [3 6 7] 7
Note:
You may assume k is always valid, 1 ≤ k ≤ input array's size for non-empty array.
Follow up:
Could you solve it in linear time?
思路 1 - 时间复杂度: O(Nk)- 空间复杂度: O(1)*****
就暴力啊,beats 17.25%
class Solution(object):
def maxSlidingWindow(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
if not nums or len(nums) == 0:
return []
res = []
for i in range(len(nums)-k+1):
res.append(max(nums[i:i+k]))
return res
Could you solve it in linear time?
思路 1 - 时间复杂度: O(N)- 空间复杂度: O(1)******
直接用deque, 照搬大神解法
# First traversing through K in the nums and only adding maximum value's index to the deque.
# Note: We are olny storing the index and not the value.
# Now, Comparing the new value in the nums with the last index value from deque,
# and if new valus is less, we don't need it
# Here we will have deque with index of maximum element for the first subsequence of length k.
# Now we will traverse from k to the end of array and do 4 things
# 1. Appending left most indexed value to the result
# 2. Checking if left most is still in the range of k (so it only allows valid sub sequence)
# 3. Checking if right most indexed element in deque is less than the new element found, if yes we will remove it
# 4. Append i at the end of the deque (Not: 3rd and 4th steps are similar to previous for loop)
beats 92.08%
class Solution(object):
def maxSlidingWindow(self, nums, k):
"""
:type nums: List[int]
:type k: int
:rtype: List[int]
"""
if not nums or len(nums) == 0:
return []
deq, res = collections.deque(), []
for i in range(k):
while deq:
if nums[i] > nums[deq[-1]]:
deq.pop()
else:
break
deq.append(i)
for i in range(k, len(nums)):
res.append(nums[deq[0]])
if deq[0] < i - k + 1:
deq.popleft()
while deq:
if nums[i] > nums[deq[-1]]:
deq.pop()
else:
break
deq.append(i)
res.append(nums[deq[0]])
return res