二叉树的统一迭代遍历

核心思想:给需要"收割"的节点贴上标签

统一迭代法的精髓在于引入了一个特殊的标记(null 指针)

在遍历过程中,我们会遇到两种状态的节点:

  1. 第一次路过(要排队):只看到了这个节点,还没完全展开它的左右子树。
  2. 第二次遇到(要收割):它的左右子树已经被处理(或按顺序排好了),现在轮到把它加入最终结果里了。

null 标记的作用,相当于一张"免死金牌"或"收割通行证"。 我们在把"当前节点(中节点)"压回栈里等待后续处理时,紧接着在它上面压入一个 null。 下次只要我们从栈里弹出一个 null,就知道:"哦,排在 null 下面的这个节点,已经走完过场了,可以直接把它的值加进结果集了!"


代码拆解:只需区分"排队"和"收割"

整个 while 循环内部,其实只有极度清晰的两大块逻辑:

java 复制代码
if (node != null) {
    // 【排队指挥官】:只负责按规定顺序把节点和孩子重新放回栈里,绝对不碰真实数据!
} else {
    // 【无脑收割机】:看到 null 标签了?拔出下一个节点,直接拿走它的值!
}
  • if 分支 负责排队部署 :弹出当前节点,然后将右节点、左节点、中节点按特定顺序重新压栈,并在中节点压栈后立马跟上一个 null
  • else 分支 负责真正收割 :一旦遇到 null,说明栈里的顺序已经排好了,立刻弹掉 null,接着弹出底下真正等待输出的节点,加入结果集(result.add)。

压栈顺序

既然明白了框架,那前序、中序、后序的区别到底在哪? 答案是:只有压栈的排队顺序不同。

因为栈是 "后进先出" 的数据结构,所以你希望节点按什么顺序被拿出 ,压栈时就必须反过来压 !并且记住一个铁律:null 永远紧紧贴在需要收割的"中节点"后面压入!

1. 前序遍历(期望结果:中 -> 左 -> 右)

反向操作,压栈顺序必须是:右 -> 左 -> 中

java 复制代码
// 前序遍历的排队逻辑
stack.pop(); 
if (node.right != null) stack.push(node.right);  // 压 右
if (node.left != null) stack.push(node.left);    // 压 左
stack.push(node);                                // 压 中
stack.push(null);                                // 紧跟标记!

2. 中序遍历(期望结果:左 -> 中 -> 右)

反向操作,压栈顺序必须是:右 -> 中 -> 左

java 复制代码
// 中序遍历的排队逻辑
stack.pop(); 
if (node.right != null) stack.push(node.right);  // 压 右
stack.push(node);                                // 压 中
stack.push(null);                                // 紧跟标记!
if (node.left != null) stack.push(node.left);    // 压 左

3. 后序遍历(期望结果:左 -> 右 -> 中)

反向操作,压栈顺序必须是:中 -> 右 -> 左

java 复制代码
// 后序遍历的排队逻辑
stack.pop(); 
stack.push(node);                                // 压 中
stack.push(null);                                // 紧跟标记!
if (node.right != null) stack.push(node.right);  // 压 右
if (node.left != null) stack.push(node.left);    // 压 左

完整源码(以中序遍历为例)

感受一下这套模板的整洁与优美:

java 复制代码
public List<Integer> inorderUnified(TreeNode root) {
    List<Integer> result = new ArrayList<>();
    Stack<TreeNode> stack = new Stack<>();
    if (root != null) stack.push(root);
    
    while (!stack.isEmpty()) {
        TreeNode node = stack.peek(); // 先看一眼栈顶是普通节点还是 null
        
        if (node != null) {
            // 【排队指挥官】
            stack.pop(); // 弹出该节点,避免重复操作
            if (node.right != null) stack.push(node.right);  // 右
            stack.push(node);                                // 中
            stack.push(null);                                // 贴上标记
            if (node.left != null) stack.push(node.left);    // 左
        } else {
            // 【无脑收割机】
            stack.pop();            // 弹出 null 标记
            node = stack.pop();     // 取出下面那个带着标记的真实节点
            result.add(node.val);   // 收割!加入结果集
        }
    }
    return result;
}

总结

这就是二叉树的统一迭代法:利用反向压栈理清顺序,利用 null 标记区分"排队"与"收割"

相关推荐
做人求其滴12 小时前
面试经典 150 题 380 274
c++·算法·面试·职场和发展·力扣
小江的记录本13 小时前
【Java基础】Java 8-21新特性:JDK21 LTS:虚拟线程、模式匹配switch、结构化并发、序列集合(附《思维导图》+《面试高频考点清单》)
java·数据库·python·mysql·spring·面试·maven
daad77713 小时前
记一组无人机IMU传感器数据
算法
计算机安禾13 小时前
【c++面向对象编程】第42篇:模板特化与偏特化:为特定类型定制实现
开发语言·c++·算法
小O的算法实验室13 小时前
2026年KBS,流形感知强化学习差分进化算法+不规则3D无人机路径规划,深度解析+性能实测
算法·智能算法·智能算法改进
玖釉-13 小时前
C++ 中的循环语句详解:while、do...while、for、嵌套循环与循环控制
开发语言·c++·算法
不做无法实现的梦~13 小时前
运动控制系统复习一览-----常考题目总结版本
算法
二宝哥13 小时前
离线安装maven
java·数据库·maven
小短腿的代码世界13 小时前
信号路由风暴:Qt算法交易系统的高频信号分发架构
qt·算法·架构
阿文的代码库13 小时前
一文读懂GROUP BY 1,2 VS GROUP BY column_1, column_2 的区别
算法