十分钟学会二叉树后序遍历循环实现


前言

开头先分享我一个朋友参加某互联网中厂的一面的算法题经历。面试官先是给了一道链表翻转的题目,我朋友思索了片刻便说不会(他刷题刷得确实不多 ),但面试官耐心引导了近20分钟,可惜最后我朋友还是没写出来。

最后没办法,面试官让我朋友写一下二叉树的后序遍历的循环实现,可以看得出来面试官是真的喜欢我这朋友,但是很可惜,我那朋友也没写出来。

我朋友一直给我说二叉树的前序遍历中序遍历循环实现 他都会,唯独后序遍历不会,他也去网上找了一些后序遍历循环实现的讲解文章,发现确实后序遍历在实现上是要复杂一点,因为大多数文章是这么说的。

后序遍历的根节点会经历两次出栈,第一次出栈是为了遍历右子树,第二次出栈进行根节点打印,所以需要记录根节点的状态。

上面的这种后序遍历的循环实现,是目前网上大多数的实现方式,也就是先遍历左子树,遍历左子树之前,根节点入栈(第一次入栈 ),左子树遍历完后,根节点出栈(第一次出栈 ),通过根节点找到右子树,在遍历右子树之前,根节点入栈(第二次入栈 ),右子树遍历完后,根节点出栈(第二次出栈),打印根节点。

其实后序遍历的循环实现有一种更简单且更易于理解的方式,突出一个优雅 ,本文将占用屏幕前的你十分钟的时间,一起来看一下如何简单的完成二叉树后序遍历的循环实现

本文所使用的二叉树节点对象如下所示。

java 复制代码
public class TreeNode {

    public int val;
    public TreeNode left;
    public TreeNode right;

    public TreeNode() {
        
    }

    public TreeNode(int x) {
        this.val = x;
    }

}

正文

首先回顾一下前序遍历 ,前序遍历的遍历顺序是根节点 -> 左子树 -> 右子树,那么前序遍历的循环实现的步骤就如下所示。

步骤一:根节点入栈;
步骤二:如果栈非空则弹出一个节点,否则结束;
步骤三:打印弹出的节点;
步骤四:如果弹出节点的右节点非空,则右节点入栈;
步骤五:如果弹出节点的左节点非空,则左节点入栈;
步骤六:循环步骤二到步骤五。

上述步骤的代码实现如下。

java 复制代码
public static void doTraversalInLoop(TreeNode treeNode) {
    if (treeNode == null) {
        return;
    }

    Stack<TreeNode> treeNodeStack = new Stack<>();
    treeNodeStack.push(treeNode);

    while (treeNodeStack.size() != 0) {
        TreeNode rootNode = treeNodeStack.pop();
        // 打印根节点
        System.out.print(rootNode.val + " ");
        // 先压右
        if (rootNode.right != null) {
            treeNodeStack.push(rootNode.right);
        }
        // 再压左
        if (rootNode.left != null) {
            treeNodeStack.push(rootNode.left);
        }
    }
}

我们可以发现,每弹出一个节点,就会打印这个节点,所以 节点弹出顺序,就是打印顺序,也就是我们的遍历顺序

在上述前序遍历的循环实现中,每次都是 先弹出根节点 ,然后 先压右子节点再压左子节点后压栈的先出栈 ),所以弹出顺序最终就是 根节点 -> 左子节点 -> 右子节点

如果上面的内容已经理解完毕,那么现在我们这样考虑,还是 先弹出根节点 ,但是 先压左子节点再压右子节点后压栈的先出栈 ),所以弹出顺序最终就成了 根节点 -> 右子节点 -> 左子节点 ,此时我们额外使用一个栈结构记作store ,每次弹出的节点就压到store 中,当所有节点遍历完毕后,我们依次弹出store 中的节点并打印,此时store 的弹出顺序就是 左子节点 -> 右子节点 -> 根节点,这个就是后序遍历了。

可以结合下面代码进行理解。

java 复制代码
public static void doTraversalInLoop(TreeNode treeNode) {
    if (treeNode == null) {
        return;
    }

    Stack<TreeNode> stack = new Stack<>();
    stack.push(treeNode);
    Stack<TreeNode> store = new Stack<>();

    while (!stack.isEmpty()) {
        TreeNode rootNode = stack.pop();
        // 将根节点收集起来
        store.push(rootNode);
        // 先压左
        if (rootNode.left != null) {
            stack.push(rootNode.left);
        }
        // 再压右
        if (rootNode.right != null) {
            stack.push(rootNode.right);
        }
    }

    while (!store.isEmpty()) {
        System.out.print(store.pop().val + " ");
    }
}

上述实现,代码十分的简洁,而且整体思想也很简单,就算是背,都能轻易背下来。

总结

核心思路就是我们通过对前序遍历的循环实现进行改造,将节点弹出顺序,由原本前序遍历的 根节点 -> 左子节点 -> 右子节点 ,变为 根节点 -> 右子节点 -> 左子节点 ,然后每个弹出的节点,我们再压到一个额外的栈结构中,当这个额外栈结构弹出节点时,顺序就和入栈顺序是反的,也就成了 左子节点 -> 右子节点 -> 根节点,而这也正是后序遍历的顺序。

分享不易,如果觉得本文对你有帮助,烦请点赞,收藏加关注,谢谢帅气漂亮的你。


相关推荐
小码的头发丝、几秒前
Spring Boot 注解
java·spring boot
java亮小白19975 分钟前
Spring循环依赖如何解决的?
java·后端·spring
飞滕人生TYF11 分钟前
java Queue 详解
java·队列
VertexGeek16 分钟前
Rust学习(八):异常处理和宏编程:
学习·算法·rust
石小石Orz17 分钟前
Three.js + AI:AI 算法生成 3D 萤火虫飞舞效果~
javascript·人工智能·算法
武子康33 分钟前
大数据-230 离线数仓 - ODS层的构建 Hive处理 UDF 与 SerDe 处理 与 当前总结
java·大数据·数据仓库·hive·hadoop·sql·hdfs
武子康35 分钟前
大数据-231 离线数仓 - DWS 层、ADS 层的创建 Hive 执行脚本
java·大数据·数据仓库·hive·hadoop·mysql
苏-言41 分钟前
Spring IOC实战指南:从零到一的构建过程
java·数据库·spring
界面开发小八哥1 小时前
更高效的Java 23开发,IntelliJ IDEA助力全面升级
java·开发语言·ide·intellij-idea·开发工具
草莓base1 小时前
【手写一个spring】spring源码的简单实现--容器启动
java·后端·spring