讲师中心 微信公众号

工具推荐

视频效率加速
Seko
Seko

AI短剧一键生成成片

Laper
Laper

剧本策划到成片协作

AniShort
AniShort

AI短剧协同制作平台

updream

一键生成高质量推广成片

如何将递归后序遍历转换为等效的迭代实现(保持 O(H) 空间复杂度)

云婷酱_9410

云婷酱_9410

发布时间:2026-11-12 11:22:03

|

340人浏览过

|

来源于php中文网

原创

如何将递归后序遍历转换为等效的迭代实现(保持 O(H) 空间复杂度)

本文详解如何将树的递归后序遍历(及依赖子树返回值的递归逻辑,如“转为累加树”)安全转化为迭代版本,通过双栈协同模拟调用栈行为,在不使用哈希表的前提下严格维持 o(h) 空间复杂度。

本文详解如何将树的递归后序遍历(及依赖子树返回值的递归逻辑,如“转为累加树”)安全转化为迭代版本,通过双栈协同模拟调用栈行为,在不使用哈希表的前提下严格维持 o(h) 空间复杂度。

在将递归算法转为迭代时,核心挑战不是“遍历顺序”,而是精确复现递归调用栈中隐含的状态流转:每个递归调用不仅需要访问节点,还需在子调用返回后,用其结果(如左/右子树和)更新当前节点。Java 虚拟机的调用栈天然保存了「当前执行位置」和「局部变量」;而纯迭代需显式建模这两者。

以题目中的 toSumTree 为例,递归函数 solve(node) 的语义是:

  • 若 node == null,返回 0;
  • 否则先递归求得 leftSum = solve(node.left) 和 rightSum = solve(node.right);
  • 再更新 node.data = leftSum + rightSum;
  • 最后返回 node.data + leftSum + rightSum(即以该节点为根的整棵子树节点值总和)。

该过程本质是后序遍历 + 子树结果聚合,不能仅靠访问序列(如 postorderTraversal 返回列表),而必须在每个节点被“处理完成”(即左右子树均已计算完毕)时,能立即获取其左右子树的返回值。

✅ 正确策略:双栈模拟调用栈状态

单栈无法区分「刚压入待处理」和「已处理完子树待收尾」两种状态。解决方案是引入双栈:

  • nodeStack:存储待访问或待收尾的节点;
  • isReturnStack(布尔栈):标记栈顶节点所对应的操作类型 — false 表示“需先处理其子树”,true 表示“子树已处理完毕,现在应回收其返回值并执行当前层逻辑”。

其设计思想直接映射递归展开:

// 递归伪码
int solve(Node node) {
    if (node == null) return 0;
    int left = solve(node.left);   // ← 压入 node.left,标记为 false(需递归)
    int right = solve(node.right); // ← 压入 node.right,标记为 false
    int result = node.data + left + right;
    node.data = left + right;      // ← 当前层计算
    return result;                 // ← 返回 result 给父调用
}

对应迭代逻辑如下(完整 toSumTree 迭代版):

class Solution {
    public void toSumTree(Node root) {
        if (root == null) return;

        Stack<Node> nodeStack = new Stack<>();
        Stack<Boolean> isReturnStack = new Stack<>(); // true: 已处理子树,可回收结果

        // 初始:压入根节点,标记为 false(需先处理子树)
        nodeStack.push(root);
        isReturnStack.push(false);

        // 存储每个节点的子树和(模拟递归返回值),复用节点字段或额外栈
        // ✅ 关键:用另一个栈 valueStack 存储每层返回值,与 nodeStack 同步进出
        Stack<Integer> valueStack = new Stack<>();

        while (!nodeStack.isEmpty()) {
            Node node = nodeStack.pop();
            boolean isReturn = isReturnStack.pop();

            if (isReturn) {
                // 此时 valueStack 栈顶两元素即为 node.right 和 node.left 的返回值(后序:右先入、左后入 → 左先出)
                int rightSum = valueStack.isEmpty() ? 0 : valueStack.pop();
                int leftSum = valueStack.isEmpty() ? 0 : valueStack.pop();

                int oldData = node.data;
                node.data = leftSum + rightSum;
                valueStack.push(oldData + leftSum + rightSum); // 返回给父节点的值
            } else {
                // 需处理子树:按后序逆序压入(右→左→当前),但当前节点标记为 true(收尾)
                // 注意:null 节点不压入,直接视为返回 0
                if (node.right != null) {
                    nodeStack.push(node.right);
                    isReturnStack.push(false);
                }
                if (node.left != null) {
                    nodeStack.push(node.left);
                    isReturnStack.push(false);
                }
                // 当前节点留到最后处理:压入并标记为 true
                nodeStack.push(node);
                isReturnStack.push(true);
                // 占位:后续将被 leftSum/rightSum 替换,此处可压 0 或忽略(由 valueStack 空判断)
            }
        }
    }
}

⚠️ 注意事项与关键点

  • 空间严格 O(H):三个栈(nodeStack, isReturnStack, valueStack)的最大深度均不超过树高 H。无哈希表、无额外节点字段,符合要求。
  • 压栈顺序决定处理顺序:后序迭代中,“当前节点”必须最后压入(且标记 true),而子节点按「右→左」顺序压入(保证左子树先被弹出处理),这是复现递归调用顺序的关键。
  • valueStack 的配对机制:每次 isReturn == true 时,从 valueStack 弹出的两个值,严格对应刚刚处理完的左右子树返回值 — 因为它们是在子节点被标记 true 后、计算完成时压入的。
  • null 处理更简洁:递归中 null 返回 0,迭代中跳过 null 节点压栈,valueStack 在弹出时用 isEmpty() 判断补 0,逻辑一致。
  • 通用性:此双栈+值栈模式适用于所有「后序型」递归,只要递归函数有明确返回值且依赖子问题解(如树直径、最大路径和、平衡判断等)。

✅ 总结

将递归转为等价迭代的核心,不是“改写遍历”,而是状态机建模:用栈显式保存「执行上下文」(节点+控制流标记+返回值)。对于后序场景,nodeStack + isReturnStack 构成控制流栈,valueStack 构成数据栈,三者协同即可 1:1 模拟 JVM 调用栈行为,既保证正确性,又守住 O(H) 空间底线。掌握这一范式,便掌握了递归转迭代的底层钥匙。

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

热门AI工具

更多
蛙蛙写作

一款AI论文写作工具,主要用于超级AI智能写作助手,适合需要提升相关任务效率的用户。

Atoms
Atoms Hot

Atoms是一款AI智能体工具,第一支自动构建真实业务的 AI 团队。

码上飞
码上飞 Hot

一款基于自然语言的AI应用开发平台,可根据用户描述自动生成应用程序,帮助个人和团队快速完成应用原型与开发。

豆包大模型

豆包大模型是一款由字节跳动推出的企业级大语言模型服务平台。

WorkBuddy

一款AI办公效率工具,主要用于腾讯云推出的AI原生桌面智能体工作台,适合需要提升相关任务效率的用户。

咔片AIPPT

一款在线AI演示文稿制作工具,可根据主题和内容需求辅助生成PPT结构与页面,提高演示材料制作效率。

Lovart
Lovart Hot

一款面向视觉设计创作的AI设计平台,可通过智能体和画布工作流辅助制作海报、Logo、网页、PPT及其他视觉内容。

花生AI
花生AI Hot

一款AI视频创作工具,主要用于B站推出的AI视频创作工具,适合需要提升相关任务效率的用户。

DeepSeek

DeepSeek是一款面向对话、写作、编程和推理场景的AI大模型工具。

相关专题

更多
Postman身份认证怎么配置
Postman身份认证怎么配置

本专题从实际API接口调试场景出发,讲解Postman身份认证的完整配置流程,包括常见认证协议介绍、授权参数设置、访问令牌填写、OAuth 2.0授权及认证失败排查,帮助用户正确完成接口授权与访问验证。

120

2026.11.06

Postman请求头Headers怎么设置
Postman请求头Headers怎么设置

围绕Postman HTTP请求头配置展开,讲解Headers选项的具体使用方式,包括Key和Value参数填写、请求头信息添加、常见Header字段设置及请求头修改方法,帮助用户解决接口调试过程中遇到的请求头配置问题。

60

2026.11.06

Postman接口请求发送教程
Postman接口请求发送教程

详细介绍Postman接口请求的创建与发送方法,涵盖GET和POST请求、URL配置、请求参数、Headers请求头、Body请求体设置以及响应结果查看,帮助用户掌握Postman接口请求的完整操作流程。

80

2026.11.06

DBeaver怎么导入数据
DBeaver怎么导入数据

DBeaver数据导入专题,整理Excel、CSV、SQL、TXT等常见文件的数据导入方法,涵盖目标数据库选择、数据表创建、字段映射、数据类型设置及导入参数配置等操作,帮助用户将本地文件中的数据写入数据库,并了解批量导入、重复数据处理和导入失败等常见问题的解决方法。

60

2026.11.05

DBeaver安全防护教程大全
DBeaver安全防护教程大全

DBeaver安全防护专题,围绕数据库连接安全、账号密码保护和访问权限管理展开,整理SSL/TLS加密连接、SSH隧道、数据库凭据保存及证书验证等配置方法,同时介绍敏感信息保护、远程连接风险和高权限账号使用等注意事项,帮助用户提高数据库日常管理的安全性。

60

2026.11.05

DBeaver连接创建与配置教程
DBeaver连接创建与配置教程

本专题聚焦DBeaver数据库连接配置与管理,覆盖从新建连接、选择数据库类型到填写连接参数和验证连接的完整流程。针对MySQL、PostgreSQL、SQL Server等常见数据库,整理相应的连接设置方法,并提供驱动缺失、认证失败和连接异常等问题的排查方向。

60

2026.11.04

XORM框架结构体映射与表结构同步
XORM框架结构体映射与表结构同步

本专题围绕XORM框架的结构体映射展开,说明表名与字段的命名策略、xorm标签常用属性、主键自增与时间字段自动填充,并演示Sync2自动建表、增量字段升级、索引声明与反向生成结构体的完整流程。

60

2026.11.04

XORM框架安装配置与数据库连接
XORM框架安装配置与数据库连接

本专题聚焦XORM框架的配置与连接环节,拆解MySQL、PostgreSQL、SQLite、SQL Server的DSN写法与驱动选择,讲清连接池参数、多数据源拆分、读写分离与超时重连的处理方式,并补充容器与兼容数据库的适配经验。

60

2026.11.04

XORM框架安装配置与数据库连接
XORM框架安装配置与数据库连接

本专题聚焦XORM框架的配置与连接环节,拆解MySQL、PostgreSQL、SQLite、SQL Server的DSN写法与驱动选择,讲清连接池参数、多数据源拆分、读写分离与超时重连的处理方式,并补充容器与兼容数据库的适配经验。

0

2026.11.04

热门下载

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

精品课程

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

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn