-
Notifications
You must be signed in to change notification settings - Fork 9
Expand file tree
/
Copy pathsliding-window-maximum.py
More file actions
66 lines (53 loc) · 1.88 KB
/
sliding-window-maximum.py
File metadata and controls
66 lines (53 loc) · 1.88 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
'''
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?
'''
class Solution:
def maxSlidingWindow(self, nums: List[int], k: int) -> List[int]:
# Approach one O(n) ~ O(kn), O(1)
# length = len(nums)
# if k > length or k <= 0: return []
# l , r = 0 , k-1
# res = [max(nums[l:r+1])]
# r += 1
# while r < length:
# if nums[r] >= res[-1]:
# res.append(nums[r])
# l += 1
# r += 1
# elif nums[l] == res[-1]:
# l += 1
# res.append(max(nums[l:r+1]))
# r += 1
# else:
# res.append(res[-1])
# l += 1
# r += 1
# return res
# Approch two 一个双向队列 O(n) , O(1)
res , tmp = [] , []
if k > len(nums) or k <= 0 : return []
for i in range(len(nums)):
if len(tmp) > 0 and tmp[0] <= i - k:
tmp.pop(0)
while len(tmp) > 0 and nums[tmp[-1]] <= nums[i]:
tmp.pop()
tmp.append(i)
if i >= k-1:
res.append(nums[tmp[0]])
return res