python滑动窗口最大值
时间: 2023-08-28 13:08:02 浏览: 79
可以使用双端队列(deque)来实现滑动窗口最大值的问题,具体步骤如下:
1. 定义一个双端队列q,用来存储窗口中的元素下标;
2. 遍历数组nums,每次将队列尾部小于当前元素的下标弹出;
3. 将当前元素的下标加入队列尾部;
4. 判断队列头部的下标是否在当前窗口中,如果不在则弹出;
5. 将队列头部对应的元素作为当前窗口最大值。
具体的 Python 代码实现如下:
```python
from collections import deque
def maxSlidingWindow(nums, k):
if not nums:
return []
q = deque()
res = []
for i in range(len(nums)):
# 如果队列头部下标已经不在窗口中,弹出队列头部
if q and q[0] < i - k + 1:
q.popleft()
# 将队列尾部小于当前元素的下标弹出
while q and nums[q[-1]] < nums[i]:
q.pop()
# 将当前元素下标加入队列尾部
q.append(i)
# 如果当前窗口的大小为k,则将队列头部元素作为当前窗口最大值
if i >= k - 1:
res.append(nums[q[0]])
return res
```
其中,nums表示输入的数组,k表示窗口大小,函数返回一个列表,表示每个窗口的最大值。例如:
```python
>>> nums = [1,3,-1,-3,5,3,6,7]
>>> k = 3
>>> maxSlidingWindow(nums, k)
[3, 3, 5, 5, 6, 7]
```
阅读全文