滑动窗口的最大值 python

最新推荐文章于 2025-06-18 09:00:21 发布

原创最新推荐文章于 2025-06-18 09:00:21 发布 · 292 阅读

0 ·

CC 4.0 BY-SA版权

刷题专栏收录该内容

70 篇文章

订阅专栏

本文介绍了一种求解滑动窗口中最大值的高效算法。通过使用双端队列来跟踪当前窗口内的元素，该算法能够在O(n)的时间复杂度内找到所有滑动窗口中的最大值。示例代码展示了如何实现这一算法，并给出了一组输入数组和窗口大小作为实例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

给定一个数组和滑动窗口的大小，找出所有滑动窗口里数值的最大值。例如，如果输入数组{2,3,4,2,6,2,5,1}及滑动窗口的大小3，那么一共存在6个滑动窗口，他们的最大值分别为{4,4,6,6,6,5}。

arr = [4,3,5,4,3,3,6,7]
w = 3
res = []
n = len(arr)
qmax = [] #双端队列
for i in range(n):
    while len(qmax)!=0 and arr[qmax[-1]] <= arr[i]:  #当数组不为空或者新来的比前面的大
        qmax.pop()
    qmax.append(i)
    if qmax[0] == i-w: #老大老了，需要淘汰了
        qmax.pop(0)
    if i >= w-1: #够三个以后就输出
        res.append(arr[qmax[0]])
print(res)