当前位置:首页 > 随笔 > 正文内容

二叉树统一迭代遍历:标记法实现前中后序

访客 随笔 2026年7月19日 4

递归实现二叉树遍历虽然简洁,但迭代方式更能揭示遍历的本质。传统的迭代写法中,前序与后序逻辑相近,而中序却需要借助指针辅助,风格割裂。本文介绍一种标记法,用完全一致的代码框架统一处理三种遍历顺序。

核心思想:访问与处理的分离

迭代遍历的难点在于:访问节点的时机与处理节点(输出值)的时机往往不同步。例如中序遍历需要先深入左子树,回溯后才处理当前节点。

解决方案是引入空指针标记:将待处理的节点入栈后,紧接着压入一个nullptr作为标记。当栈顶为空指针时,说明下一个节点已准备好被处理。

中序遍历实现

中序遍历的顺序为左-根-右。栈中节点的排列需保证:左子树优先访问,当前节点延迟处理。

class Solution {
public:
    vector<int> inorderTraversal(TreeNode* curr) {
        vector<int> output;
        stack<TreeNode*> stk;
        if (curr) stk.push(curr);
        
        while (!stk.empty()) {
            TreeNode* top = stk.top();
            if (top) {
                stk.pop();
                // 右子树先入栈(后访问)
                if (top->right) stk.push(top->right);
                // 当前节点标记为待处理
                stk.push(top);
                stk.push(nullptr);
                // 左子树最后入栈(先访问)
                if (top->left) stk.push(top->left);
            } else {
                // 遇到标记,处理节点
                stk.pop();
                TreeNode* ready = stk.top();
                stk.pop();
                output.push_back(ready->val);
            }
        }
        return output;
    }
};

前序遍历:调整入栈顺序

前序遍历为根-左-右。仅需改变子节点的入栈顺序:右子树先入,左子树后入,当前节点标记在最前。

class Solution {
public:
    vector<int> preorderTraversal(TreeNode* curr) {
        vector<int> output;
        stack<TreeNode*> stk;
        if (curr) stk.push(curr);
        
        while (!stk.empty()) {
            TreeNode* top = stk.top();
            if (top) {
                stk.pop();
                // 右子树先入栈
                if (top->right) stk.push(top->right);
                // 左子树后入栈
                if (top->left) stk.push(top->left);
                // 当前节点标记(最先处理)
                stk.push(top);
                stk.push(nullptr);
            } else {
                stk.pop();
                TreeNode* ready = stk.top();
                stk.pop();
                output.push_back(ready->val);
            }
        }
        return output;
    }
};

后序遍历:再次调整入栈顺序

后序遍历为左-右-根。当前节点最先入栈标记,随后右子树、左子树依次入栈。

class Solution {
public:
    vector<int> postorderTraversal(TreeNode* curr) {
        vector<int> output;
        stack<TreeNode*> stk;
        if (curr) stk.push(curr);
        
        while (!stk.empty()) {
            TreeNode* top = stk.top();
            if (top) {
                stk.pop();
                // 当前节点标记(最后处理)
                stk.push(top);
                stk.push(nullptr);
                // 右子树后访问
                if (top->right) stk.push(top->right);
                // 左子树先访问
                if (top->left) stk.push(top->left);
            } else {
                stk.pop();
                TreeNode* ready = stk.top();
                stk.pop();
                output.push_back(ready->val);
            }
        }
        return output;
    }
};

模式总结

三种遍历的代码结构完全一致,区别仅在于子节点与标记节点的入栈顺序

遍历类型入栈顺序(从先到后)
前序标记节点 → 左子树 → 右子树
中序右子树 → 标记节点 → 左子树
后序标记节点 → 右子树 → 左子树

空指针标记法将递归的隐式调用栈显式化,同时统一了迭代遍历的代码风格,便于记忆和维护。

相关文章

可以按小时收费的VPS

很多 VPS 提供商都支持 按小时计费(hourly billing),想短期试用 / 临时搭建节点、测试网络、短期项目等场景非常合适。下面是当前最主流且靠谱的按小时 VPS 选项,分别按不同需求场景整理: 1. Vultr(全球节点,包括日本) 按小时计费 可选机房:东京 / 大阪 / 洛杉矶 / 法兰克福 / 伦敦 … 支持 PayPal(部分情况),但更常用信用卡/PayPal+卡价格参考$...

在 iPhone 上下载国外App

地区/国家限制App Store 会根据 Apple ID 的国家或地区限制应用下载。如果你的 Apple ID 绑定的是中国大陆,就可能无法下载 OpenAI 官方的 ChatGPT 应用,因为它在大陆 App Store 不上架。解决办法:换成美国、加拿大、香港等地区的 Apple ID。或者在现有 Apple ID 上更改地区。注册一个国外 Apple ID(推荐)比如注册 美国区 Appl...

Node.js 中的异步编程:回调与 Promise

Node.js 是一个基于 JavaScript 构建的单线程、非阻塞运行环境,它通过异步编程机制来高效处理多个操作。在执行如文件读取、API 请求或数据库查询等任务时,Node.js 不会等待这些操作完成,而是使用回调函数和 Promise 来避免阻塞主线程。 回调方式实现异步 那么当异步操作完成后,Node.js 如何知道接下来要做什么呢?这就要用到 回调函数(callback)。 回调本质上...

Selenium自动化测试入门指南

Selenium自动化测试入门指南

什么是自动化测试? 自动化测试是指利用软件工具自动执行测试用例,模拟用户操作,如打开网页、点击链接、输入文本等,并验证结果是否符合预期。 其主要优点包括: 大幅减少人工成本 测试速度快 可以在非工作时间运行 支持持续集成和交付 然而,它也存在一些局限性,例如开发成本较高、不适合快速变化的项目、依赖稳定的UI界面等。 自动化测试的应用条件 适合引入自动化测试的情况包括: 手动测试耗时且需要大量...

MariaDB Galera集群故障快速恢复指南

OpenStack控制节点采用三节点MariaDB Galera集群架构。当数据库集群因故障重启时,有时会出现Galera集群无法正常启动的问题。虽然有多种方法可以恢复数据库服务,但如何实现快速启动同时确保数据完整性呢? 通过分析日志发现,MariaDB Galera集群节点宕机时会在日志中输出以下信息: [Note] WSREP: 新集群视图:全局状态: 874d8e7e-5980-11e8-8...

Android 中 EventBus 的通信机制与实现原理深度解析

EventBus 核心设计思想 EventBus 是一个基于观察者模式的事件总线框架,广泛应用于 Android 平台以实现组件解耦。它通过中心化的消息分发机制,使不同层级、不同线程的对象能够以"发布-订阅"方式通信,避免了传统接口回调或广播带来的强依赖问题。 核心角色说明 事件(Event):任意 Java 对象,作为数据载体,如网络状态变更通知、用户登录信息等。 发布者(Publi...

发表评论

访客

◎欢迎参与讨论,请在这里发表您的看法和观点。