0

0

使用单调栈优化Python代码的时间复杂度:一个将数字编码的案例

花韻仙語

花韻仙語

发布时间:2025-09-16 19:45:01

|

790人浏览过

|

来源于php中文网

原创

 使用单调栈优化Python代码的时间复杂度:一个将数字编码的案例

本文旨在指导读者如何使用单调栈这一数据结构,将一个时间复杂度为O(n²)的Python代码优化至O(n)。我们将通过一个具体的编码问题——将数组中的每个数字加上其后第一个更大的数字(如果不存在则加上自身)——来详细讲解单调栈的原理和应用,并提供清晰的代码示例和逐步解释。 ### 问题描述 给定一个数字数组,目标是将数组中的每个数字编码。编码规则是:对于数组中的每个数字,找到其后第一个比它大的数字,并将两者相加。如果该数字后面没有更大的数字,则将该数字与自身相加。例如,对于输入数组 `[4, 3, 7, 3, 2, 8, 6, 1, 10, 3]`,编码后的结果应该是 `[11, 10, 15, 11, 10, 18, 16, 11, 10, 3]`。 ### 原始代码及其时间复杂度分析 提供的原始代码使用队列 `queue.Queue()` 来实现编码逻辑,其核心思想是遍历队列中的每个元素,并在队列的剩余部分中查找第一个更大的元素。 ```python import queue q = queue.Queue() a = [4, 3, 7, 3, 2, 8, 6, 1, 10, 3] for i in a: q.put(i) encoded = [] while q: current = q.get() for i in range(q.qsize()): if current append(q.queue[i] + current) break print(encoded)

这段代码的时间复杂度是 o(n²),因为对于队列中的每个元素,都需要遍历队列的剩余部分来寻找更大的元素。

使用单调栈优化

单调栈是一种特殊的栈结构,其内部元素保持单调递增或单调递减的顺序。利用单调栈,我们可以高效地找到数组中每个元素后面第一个更大的元素。

算法思路:

  1. 创建一个空栈 s 用于存储数组元素的索引。
  2. 遍历数组 a,对于每个元素 x,执行以下操作:
    • 如果栈不为空,并且当前元素 x 大于栈顶元素对应的值 a[s[-1]],则循环执行以下操作:
      • 将栈顶元素弹出,并将其对应编码后的值更新为栈顶元素的值加上当前元素 x。
    • 将当前元素的索引 i 压入栈中。
  3. 遍历结束后,栈中剩余的元素表示它们后面没有更大的元素,因此它们的编码值保持不变(即加上自身,但由于初始化时已经复制了数组,所以无需额外处理)。

Python 代码实现:

a = [4, 3, 7, 3, 2, 8, 6, 1, 10, 3]

encoded = a[:]  # 创建一个a的副本,用于存放编码后的值
s = []  # 创建一个空栈
for i, x in enumerate(a):
    while s and x > a[s[-1]]:
        encoded[s.pop()] += x
    s.append(i)

print(encoded)

代码解释:

立即学习Python免费学习笔记(深入)”;

MusicLM
MusicLM

谷歌平台的AI作曲工具,用文字生成音乐

下载
  • encoded = a[:] 创建了数组 a 的一个副本,这是为了避免直接修改原始数组。
  • s = [] 初始化一个空栈,用于存储数组元素的索引。
  • for i, x in enumerate(a): 遍历数组 a,同时获取元素的索引 i 和值 x。
  • while s and x > a[s[-1]]: 这是一个循环,用于处理栈中元素。如果栈不为空,并且当前元素 x 大于栈顶元素对应的值 a[s[-1]],则说明找到了栈顶元素后面第一个更大的元素。
  • encoded[s.pop()] += x 将栈顶元素弹出,并将其对应编码后的值更新为栈顶元素的值加上当前元素 x。
  • s.append(i) 将当前元素的索引 i 压入栈中。

时间复杂度分析:

虽然代码中有一个 while 循环,但每个元素最多入栈一次,也最多出栈一次。因此,内层 while 循环的总执行次数不会超过 n,其中 n 是数组的长度。所以,整个算法的时间复杂度是 O(n)。

注意事项和总结

  • 单调栈是一种非常有用的数据结构,可以用于解决很多与寻找“下一个更大/更小元素”相关的问题。
  • 在使用单调栈时,需要仔细考虑栈中应该存储元素的值还是索引,以及如何维护栈的单调性。
  • 上述代码中,我们存储的是元素的索引,这样可以方便地更新 encoded 数组。
  • 理解单调栈的关键在于理解其单调性如何帮助我们找到目标元素,并避免不必要的比较。

通过使用单调栈,我们将原始代码的时间复杂度从 O(n²) 降低到 O(n),显著提高了代码的效率。这个案例展示了如何利用合适的数据结构和算法来优化代码的性能。

					

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

通义千问
通义千问

阿里巴巴推出的全能AI助手

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
python中print函数的用法
python中print函数的用法

python中print函数的语法是“print(value1, value2, ..., sep=' ', end=' ', file=sys.stdout, flush=False)”。本专题为大家提供print相关的文章、下载、课程内容,供大家免费下载体验。

186

2023.09.27

if什么意思
if什么意思

if的意思是“如果”的条件。它是一个用于引导条件语句的关键词,用于根据特定条件的真假情况来执行不同的代码块。本专题提供if什么意思的相关文章,供大家免费阅读。

778

2023.08.22

while的用法
while的用法

while的用法是“while 条件: 代码块”,条件是一个表达式,当条件为真时,执行代码块,然后再次判断条件是否为真,如果为真则继续执行代码块,直到条件为假为止。本专题为大家提供while相关的文章、下载、课程内容,供大家免费下载体验。

97

2023.09.25

java中break的作用
java中break的作用

本专题整合了java中break的用法教程,阅读专题下面的文章了解更多详细内容。

118

2025.10.15

java break和continue
java break和continue

本专题整合了java break和continue的区别相关内容,阅读专题下面的文章了解更多详细内容。

258

2025.10.24

treenode的用法
treenode的用法

​在计算机编程领域,TreeNode是一种常见的数据结构,通常用于构建树形结构。在不同的编程语言中,TreeNode可能有不同的实现方式和用法,通常用于表示树的节点信息。更多关于treenode相关问题详情请看本专题下面的文章。php中文网欢迎大家前来学习。

539

2023.12.01

C++ 高效算法与数据结构
C++ 高效算法与数据结构

本专题讲解 C++ 中常用算法与数据结构的实现与优化,涵盖排序算法(快速排序、归并排序)、查找算法、图算法、动态规划、贪心算法等,并结合实际案例分析如何选择最优算法来提高程序效率。通过深入理解数据结构(链表、树、堆、哈希表等),帮助开发者提升 在复杂应用中的算法设计与性能优化能力。

17

2025.12.22

深入理解算法:高效算法与数据结构专题
深入理解算法:高效算法与数据结构专题

本专题专注于算法与数据结构的核心概念,适合想深入理解并提升编程能力的开发者。专题内容包括常见数据结构的实现与应用,如数组、链表、栈、队列、哈希表、树、图等;以及高效的排序算法、搜索算法、动态规划等经典算法。通过详细的讲解与复杂度分析,帮助开发者不仅能熟练运用这些基础知识,还能在实际编程中优化性能,提高代码的执行效率。本专题适合准备面试的开发者,也适合希望提高算法思维的编程爱好者。

28

2026.01.06

java入门学习合集
java入门学习合集

本专题整合了java入门学习指南、初学者项目实战、入门到精通等等内容,阅读专题下面的文章了解更多详细学习方法。

1

2026.01.29

热门下载

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

精品课程

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

共4课时 | 22.3万人学习

Django 教程
Django 教程

共28课时 | 3.6万人学习

SciPy 教程
SciPy 教程

共10课时 | 1.3万人学习

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

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