如何使用单调栈优化 Python 代码的时间复杂度
本文旨在指导读者如何使用单调栈这一数据结构，将原本时间复杂度为O(n²)的Python代码优化至O(n)。通过具体示例和详细解释，我们将展示如何利用单调栈高效地找到数组中每个元素的下一个更大元素，从而提升算法性能。
代码解释
- encoded = a[:] 创建了输入数组 a 的一个副本，这样修改 encoded 不会影响原始数组。
- s = [] 初始化了单调栈。
- enumerate(a) 用于同时获取数组的索引和值。
- while s and x > a[s[-1]] 循环确保栈的单调性。当遇到比栈顶元素更大的元素时，不断弹出栈顶元素，直到栈为空或者栈顶元素大于等于当前元素。
- encoded[s.pop()] += x 将栈顶元素对应的值更新为与当前元素的和。
- s.append(i) 将当前元素的索引压入栈中。
复杂度分析
- 时间复杂度： O(n)。虽然代码中包含一个 while 循环，但每个元素最多入栈一次，出栈一次，因此总的时间复杂度为 O(n)。
- 空间复杂度： O(n)。单调栈最多存储 n 个元素的索引。
注意事项
- 单调栈适用于解决寻找数组中下一个更大/更小元素的问题。
- 在实际应用中，可以根据具体需求选择单调递增栈或单调递减栈。
- 使用单调栈时，需要注意维护栈的单调性，确保算法的正确性。
总结
通过使用单调栈，我们可以将原本时间复杂度为 O(n²) 的代码优化至 O(n)，显著提升算法的性能。单调栈是一种强大的数据结构，在解决与数组元素大小关系相关的问题时非常有用。理解单调栈的工作原理和应用场景，可以帮助我们编写更高效的 Python 代码。
火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力，依托字节自研豆包多模态大模型，凭借火山方舟一站式平台，提供模型精调、推理部署、智能应用搭建等全链路服务，结合高性能AI算力，助力各行业快速落地大模型数字化与智能化应用。
AirBuddy 3 是面向 macOS 的蓝牙外设增强工具，可集中查看 AirPods、Beats 与苹果设备电量，提供快速连接、播放控制、电量提醒、桌面小组件和多台 Mac 外设接力等功能。
AirServer macOS版可把 Mac 转换为无线投屏接收端，通过 AirPlay 或 Google Cast 接收手机、平板和电脑画面，并支持多设备同时连接、高清镜像及会话录制，适合家庭娱乐、课堂教学、会议演示和应用展示。
Autodesk Maya 是面向影视、电视、动画和游戏行业的专业三维制作软件，集多边形与 NURBS 建模、角色绑定、关键帧和动作捕捉动画、Bifrost 特效模拟、材质开发及 Arnold 渲染于一体，
AirServer Windows版可把电脑变成AirPlay、Google Cast和Miracast无线投屏接收端，适合会议演示、课堂教学及家庭大屏分享。发送设备通常无需另装客户端，并可同时接收和智能排列多个镜像会话。