0

0

Python单调栈怎么用_寻找下一个更大元素的线性时间解

P粉602998670

P粉602998670

发布时间:2026-03-18 07:55:32

|

984人浏览过

|

来源于php中文网

原创

单调栈O(n)找下一个更大元素,因各元素至多入栈出栈一次;栈存索引、维持递减序,遇更大数即弹栈并记录答案;环形数组用两次遍历+取模,栈中索引保持0到n-1。

python单调栈怎么用_寻找下一个更大元素的线性时间解

单调栈的核心逻辑:为什么它能 O(n) 找到下一个更大元素

因为每个元素最多入栈、出栈各一次。不是靠暴力扫后续,而是用栈维护“还没找到答案”的候选者,新元素一来,立刻清算所有比它小的旧元素——它们的答案就是当前这个新元素。

关键点在于:栈里永远保持递减(或非递增)顺序。一旦遇到更大的数,就持续弹栈,给弹出的每个元素记下当前值作为 next_greater

  • 栈中存的是索引,不是值,方便回填结果数组
  • 初始化结果数组为 -1,代表“没找到”,避免遗漏未出栈的元素
  • 遍历完后栈内剩余索引对应元素,答案就是默认的 -1

Python 实现时容易错的三个地方

写出来像模像样,但跑起来答案对不上,大概率栽在这三处:

  • 误把 stack.pop() 的返回值当索引用——它返回的是索引,但你得用它去查原数组:nums[stack.pop()] 是值,stack.pop() 本身才是要填答案的位置
  • 循环条件写成 while stack and nums[i] > nums[stack[-1]] 没问题,但漏了 stack 非空判断,导致空栈取 stack[-1]IndexError: list index out of range
  • 结果数组长度设错:必须是 len(nums),且初始全为 -1;有人用 [] append,顺序和索引就对不上了

正确片段示例:

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

SongAI
SongAI

免费AI歌曲和音乐生成平台,支持文字生成歌曲、AI歌词创作、AI翻唱等功能

下载
res = [-1] * len(nums)
stack = []
for i in range(len(nums)):
    while stack and nums[i] > nums[stack[-1]]:
        idx = stack.pop()
        res[idx] = nums[i]
    stack.append(i)

处理环形数组:% len(nums) 不是万能解药

题目变体常要求“下一个更大元素 II”,即数组首尾相接。直接在原数组上双倍长度模拟,会多算一次,且空间浪费。

更稳妥的做法是遍历两次索引,但只用一个原始数组:

  • 外层循环 i in range(2 * len(nums)),但取值用 nums[i % len(nums)]
  • 栈里仍存原始索引(0 到 n-1),避免越界
  • 只在第一次遍历时允许入栈(if i ),否则只出栈不进栈
  • 注意:结果数组只更新一次,所以检查 res[idx] == -1 再赋值,防止被第二次覆盖

什么时候不该硬套单调栈

如果需求变成“下一个更大元素的下标差”“前一个更大元素”“大于等于而非严格大于”,逻辑要微调,但栈结构还能用。真正该换思路的情况有:

  • 需要动态插入/删除中间元素——单调栈是离线算法,不支持修改
  • 查询带范围限制(比如“右侧 5 个数以内”)——此时单调栈失去优势,直接滑动窗口或线段树更合适
  • 数据量极小(

单调栈真正的价值边界很清晰:单次遍历、静态数组、全局最值关系、严格线性时间。超出这个范围,先想清楚“栈到底在维护什么秩序”,别为了用而用。

热门AI工具

更多
DeepSeek
DeepSeek

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

豆包大模型
豆包大模型

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

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

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

文心一言
文心一言

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

讯飞写作
讯飞写作

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

即梦AI
即梦AI

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

ChatGPT
ChatGPT

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

相关专题

更多
if什么意思
if什么意思

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

848

2023.08.22

while的用法
while的用法

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

108

2023.09.25

堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

448

2023.07.18

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

606

2023.08.10

堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

448

2023.07.18

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

606

2023.08.10

append用法
append用法

append是一个常用的命令行工具,用于将一个文件的内容追加到另一个文件的末尾。想了解更多append用法相关内容,可以阅读本专题下面的文章。

349

2023.10.25

python中append的用法
python中append的用法

在Python中,append()是列表对象的一个方法,用于向列表末尾添加一个元素。想了解更多append的更多内容,可以阅读本专题下面的文章。

1080

2023.11.14

抖漫入口地址合集
抖漫入口地址合集

本专题整合了抖漫入口地址相关合集,阅读专题下面的文章了解更多详细地址。

17

2026.03.17

热门下载

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

精品课程

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

共4课时 | 22.5万人学习

Django 教程
Django 教程

共28课时 | 5.1万人学习

SciPy 教程
SciPy 教程

共10课时 | 2万人学习

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

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