java----树的后序遍历(非递归)

最新推荐文章于 2022-03-31 18:52:50 发布

原创最新推荐文章于 2022-03-31 18:52:50 发布 · 764 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #stack #栈 #java

Java语言专栏收录该内容

57 篇文章

订阅专栏

本文介绍了一种非递归方式实现的树的后序遍历算法。通过使用双栈或者单栈配合标记的方式，避免了递归带来的额外开销，并详细解释了每一步的操作流程。

问题描述
实现树的后序遍历(非递归)
源代码
代码中有注释

public class TreeNode {
    public char val;
    public TreeNode left;
    public TreeNode right;
    public TreeNode(char var) {
        this.val = var;
    }
}

import java.util.Deque;
import java.util.LinkedList;

public class PostTraversal {
    public void postTraversal1(TreeNode root) {
        Deque<TreeNode> stack = new LinkedList<>();
        // 当前正在遍历的结点
        TreeNode cur = root;
        // 刚被弹出栈的结点.
        TreeNode pre = null;
        while (!stack.isEmpty() || cur != null) {
            // 当 cur 不为空时, 将左边的结点一直压栈
            while (cur != null) {
                stack.push(cur);
                cur = cur.left;
            }
            // 当 cur 为空时, 查看栈顶元素
            TreeNode top = stack.peek();
            // 查看栈顶元素是否有右节点
            // 1. 当无右节点时, 输出该节点的值, 并将该节点从栈中弹出, 此时需要记录刚被弹出的结点
            // 2. 当有右节点并且右节点没有被遍历过, 此时就将当前节点指向其右节点.
            // 3. 当有右节点时, 并且右节点已经被遍历过, 此时将该节点弹出, 记录弹出的结点.
            if (top.right == null) {
                // 无右孩子
                System.out.println(top.val);
                pre = stack.pop();
            } else if (top .right == pre) {
                // 有右孩子并且右孩子已经被遍历过
                System.out.println(top.val);
                pre = stack.pop();
            } else {
                // 右孩子没有被遍历过.
                cur = top.right;
            }
        }
    }
}