0

0

标题:用单调双端队列实现 O(n) 时间复杂度的滑动窗口最大值索引查询

花韻仙語

花韻仙語

发布时间:2026-01-09 22:08:05

|

495人浏览过

|

来源于php中文网

原创

标题:用单调双端队列实现 O(n) 时间复杂度的滑动窗口最大值索引查询

给定两个等长整数数组 `x`(元素为“能量值”)和 `k`(每个位置 `i` 定义窗口长度 `k[i]`),要求对每个 `i` 快速找出子数组 `x[i−k[i]+1 .. i]` 中最大值的**最左索引**,整体时间复杂度需严格为 Θ(n)。

该问题本质是带变长右对齐滑动窗口的最大值索引查询:窗口始终以当前下标 i 为右端点,左端点为 i − k[i] + 1(含),且 k 数组满足约束 k[i+1] ≤ k[i] + 1(即窗口长度变化平缓,不会突增)。这使得我们无法直接套用固定长度滑动窗口的经典单调队列解法,但该约束恰恰是突破口——它保证了窗口左边界随 i 增加而非递减(即左端点单调不降),因此可设计一个单次遍历 + 单调双端队列(deque) 的 Θ(n) 解法。

核心思想是:维护一个严格递减的双端队列 dq,存储的是数组 x 的下标,确保队首始终是当前有效窗口内的最大值索引。关键在于——由于 k[i] 变化受限,窗口左边界 left_i = i − k[i] + 1 是单调不减序列,因此我们可以在主循环中只做最多一次左边界收缩,避免重复扫描。

以下是完整、正确、Θ(n) 时间复杂度的 Python 实现:

from collections import deque

def max_index_for_each_k(x, k):
    n = len(x)
    result = [0] * n
    dq = deque()  # 存储下标,对应 x 值严格递减

    for i in range(n):
        # Step 1: 移除超出当前窗口左边界的所有下标
        left_bound = i - k[i] + 1
        while dq and dq[0] < left_bound:
            dq.popleft()

        # Step 2: 维护单调递减性:从队尾弹出所有 x[j] <= x[i] 的下标 j
        # 注意:此处用 <= 而非 <,是为了在值相等时保留更早(更左)的索引,
        # 从而自然满足“返回最左最大索引”的要求
        while dq and x[dq[-1]] <= x[i]:
            dq.pop()

        # Step 3: 加入当前下标
        dq.append(i)

        # Step 4: 队首即为当前窗口 [left_bound, i] 中最大值的最左索引
        result[i] = dq[0]

    return result

# 测试用例
x = [1000, 4, 3, 2, 500, 10, 1]
k = [1, 2, 2, 3, 2, 3, 1]
print(" ".join(map(str, max_index_for_each_k(x, k))))  # 输出:0 0 1 1 4 4 6

为什么这是 Θ(n)?

Content at Scale
Content at Scale

SEO长内容自动化创作平台

下载
  • 每个下标 i 最多入队 1 次、出队 1 次(popleft() 和 pop() 总次数 ≤ 2n);
  • left_bound 单调不减,因此 popleft() 在整个循环中总执行次数 ≤ n;
  • 无嵌套循环,主循环 for i in range(n) 为 O(n),内部操作均摊 O(1)。

⚠️ 关键注意事项

  • k[i] 必须满足题目约束 k[i+1] ≤ k[i] + 1(且 k[0] ≥ 1),否则左边界可能回跳,破坏单调性,导致算法失效;
  • 使用 保留最早出现的下标(即最左索引),这与题意“返回最强大索引(即最大值的首次出现位置)”完全一致;
  • 不要使用 x.index(max(...)) 或类似线性查找,那会退化为 O(n·k);
  • 段树或稀疏表虽可解一般 RMQ,但时间复杂度为 O(n log n) 预处理 + O(log n) 查询,总复杂度 O(n log n),不满足本题 Θ(n) 要求。

综上,利用 k 数组的特殊单调性约束,结合单调双端队列维护“右对齐变长窗口”的最大值索引,是实现真正线性时间解法的唯一高效路径。

相关专题

更多
python开发工具
python开发工具

php中文网为大家提供各种python开发工具,好的开发工具,可帮助开发者攻克编程学习中的基础障碍,理解每一行源代码在程序执行时在计算机中的过程。php中文网还为大家带来python相关课程以及相关文章等内容,供大家免费下载使用。

741

2023.06.15

python打包成可执行文件
python打包成可执行文件

本专题为大家带来python打包成可执行文件相关的文章,大家可以免费的下载体验。

634

2023.07.20

python能做什么
python能做什么

python能做的有:可用于开发基于控制台的应用程序、多媒体部分开发、用于开发基于Web的应用程序、使用python处理数据、系统编程等等。本专题为大家提供python相关的各种文章、以及下载和课程。

756

2023.07.25

format在python中的用法
format在python中的用法

Python中的format是一种字符串格式化方法,用于将变量或值插入到字符串中的占位符位置。通过format方法,我们可以动态地构建字符串,使其包含不同值。php中文网给大家带来了相关的教程以及文章,欢迎大家前来阅读学习。

617

2023.07.31

python教程
python教程

Python已成为一门网红语言,即使是在非编程开发者当中,也掀起了一股学习的热潮。本专题为大家带来python教程的相关文章,大家可以免费体验学习。

1259

2023.08.03

python环境变量的配置
python环境变量的配置

Python是一种流行的编程语言,被广泛用于软件开发、数据分析和科学计算等领域。在安装Python之后,我们需要配置环境变量,以便在任何位置都能够访问Python的可执行文件。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

547

2023.08.04

python eval
python eval

eval函数是Python中一个非常强大的函数,它可以将字符串作为Python代码进行执行,实现动态编程的效果。然而,由于其潜在的安全风险和性能问题,需要谨慎使用。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

577

2023.08.04

scratch和python区别
scratch和python区别

scratch和python的区别:1、scratch是一种专为初学者设计的图形化编程语言,python是一种文本编程语言;2、scratch使用的是基于积木的编程语法,python采用更加传统的文本编程语法等等。本专题为大家提供scratch和python相关的文章、下载、课程内容,供大家免费下载体验。

705

2023.08.11

c++主流开发框架汇总
c++主流开发框架汇总

本专题整合了c++开发框架推荐,阅读专题下面的文章了解更多详细内容。

3

2026.01.09

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
最新Python教程 从入门到精通
最新Python教程 从入门到精通

共4课时 | 0.6万人学习

Django 教程
Django 教程

共28课时 | 2.9万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.1万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号