滑动窗口最大值python
时间: 2023-08-30 20:01:10 浏览: 216
滑动窗口最大值问题,可以使用双端队列来解决。双端队列中存放的是窗口中的元素的索引值,而且队列中的元素按照从大到小的顺序排列。具体算法如下:
1. 创建一个双端队列 deque,用来存放窗口中的元素的索引值。
2. 初始化结果数组 res,用来存放每个窗口中的最大值。
3. 遍历整个数组,做如下操作:
a. 检查队列中的头部元素是否已经越过窗口的范围,如果是,则删除头部元素。
b. 判断当前元素是否比队列中的尾部元素大,如果是,则将尾部元素删除,直到队列为空或者队列中的元素比当前元素大。
c. 将当前元素的索引加入队列的尾部。
d. 判断当前窗口是否已经形成,如果是,则将队列中的头部元素加入结果数组 res。
4. 返回结果数组 res。
下面是用 Python 实现的代码:
```python
from collections import deque
def slidingWindowMax(nums, k):
deque = deque()
res = []
for i in range(len(nums)):
if deque and i - deque[0] == k:
deque.popleft()
while deque and nums[i] > nums[deque[-1]]:
deque.pop()
deque.append(i)
if i >= k - 1:
res.append(nums[deque[0]])
return res
```
以上就是滑动窗口最大值问题的 Python 解法。该算法的时间复杂度为 O(n),其中 n 是数组的长度。
阅读全文