lintcode68-二叉树的后序遍历

最新推荐文章于 2022-03-19 19:48:40 发布

小学生_

最新推荐文章于 2022-03-19 19:48:40 发布

阅读量217

点赞数

CC 4.0 BY-SA版权

分类专栏： lintcode lintcode-二叉树

本文链接：https://blog.youkuaiyun.com/xiaowei132/article/details/88374184

lintcode 同时被 2 个专栏收录

12 篇文章

订阅专栏

lintcode-二叉树

12 篇文章

订阅专栏

本文详细解析了二叉树后序遍历的算法实现，通过使用栈结构，巧妙地实现了在访问完左右子树后再访问根节点的过程。文章提供了具体的代码示例，帮助读者理解算法流程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Question：

给出一棵二叉树，返回其节点值的后序遍历。
给出一棵二叉树 {1,#,2,3},
1
\
2
/
3
返回 [3,2,1]

Solution：
二叉树的遍历只要记住
1、借助栈
2、前序遍历：入栈时访问
3、中序遍历：弹出栈时访问

4、后续遍历：关键点在于，要先访问完左右子树后，才能访问当前节点
那么就用一个pre节点标记刚访问过的节点，如果当前节点的右儿子为空或者右儿子是刚刚访问过的节点，此时弹出并访问栈顶元素

public List<Integer> postorderTraversal(TreeNode root) {
        // write your code here
        List<Integer> result = new ArrayList<Integer>();
        if(root == null)
            return result;
        TreeNode cur = root;
        TreeNode pre = null;
        Stack<TreeNode> stack = new Stack<TreeNode>();
        while(cur != null || !stack.isEmpty()) {
            //所有左子树压入站
            if(cur != null) {
                stack.push(cur);
                cur = cur.left;
            } else {
                //左子树压入完成，获取栈顶元素
                cur = stack.peek();
                //右子树为空或右子树已经访问过了
                if(cur.right == null || cur.right == pre) {
                    cur = stack.pop();
                    result.add(cur.val);
                    pre = cur;
                    //cur置空，避免cur再次入栈导致循环
                    cur = null;
                } else {
                    //有右儿子没被访问，在下一次循环时入栈
                    cur = cur.right;
                }
            }
        }
        return result;
    }

    @Test
    public void test() {
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(2);
        root.left.right = new TreeNode(5);
        root.left.left = new TreeNode(4);
        root.right = new TreeNode(3);
        List<Integer> result = postorderTraversal(root);
        System.out.println(result.toString());
    }
}