二叉树后序遍历（递归、迭代）

学海无涯苦作舟呀

已于 2022-03-29 22:20:33 修改

阅读量636

点赞数

分类专栏： CodeTop企业题库——数据研发岗 # 二叉树文章标签： leetcode dfs 算法

于 2022-03-29 22:08:42 首次发布

本文链接：https://blog.csdn.net/weixin_46184836/article/details/123831074

版权

CodeTop企业题库——数据研发岗同时被 2 个专栏收录

47 篇文章 8 订阅

订阅专栏

二叉树

4 篇文章 1 订阅

订阅专栏

本文详细解析了二叉树后序遍历的递归和迭代两种方法，结合LeetCode上的题目，介绍了解题思路。在后序遍历中，由于需要确保先访问左右子树，再访问根节点，因此在迭代过程中引入了(prev)来辅助判断节点的访问状态。

摘要由CSDN通过智能技术生成

题目链接：https://leetcode-cn.com/problems/binary-tree-postorder-traversal/

给定一个二叉树的根节点 root ，返回它的后序遍历。
代码如下：

1.递归法

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        postorder(root, res);
        return res;
    }

    public void postorder(TreeNode root, List<Integer> res) {
        if (root == null) {
            return;
        }
        postorder(root.left, res);//左
        postorder(root.right, res);//右
        res.add(root.val);//根
    }
}

2.迭代法

解题思路：
与中序的不同之处在于：中序遍历中，从栈中弹出的节点，其左子树是访问完了，可以直接访问该节点，然后接下来访问右子树。

后序遍历中，从栈中弹出的节点，我们只能确定其左子树肯定访问完了，但是无法确定右子树是否访问过。
因此，我们在后序遍历中，引入了一个prev来记录历史访问记录。

当访问完一棵子树的时候，我们用prev指向该节点。
这样，在回溯到父节点的时候，我们可以依据prev是指向左子节点，还是右子节点，来判断父节点的访问情况。

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        if (root == null) {
            return res;
        }

        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode prev = null;
        while (root != null || !stack.isEmpty()) {
            while (root != null) {
                stack.push(root);
                root = root.left;
            }
            root = stack.pop();
            if (root.right == null || root.right == prev) {
                res.add(root.val);
                prev = root;
                root = null;
            } else {
                stack.push(root);
                root = root.right;
            }
        }
        return res;
    }
}