【LeetCode刷题】145 二叉树的后序遍历 java

题目

给你一棵二叉树的根节点 root ，返回其节点值的 后序遍历 。

示例

方法一：递归

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);
    }
}

 方法二：迭代

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        if(root == null)
            return res;
        Deque<TreeNode> stack = new LinkedList<TreeNode>();
        TreeNode prev = null;   //用来判断右孩子有没有被访问
        while(root != null || !stack.isEmpty()){
            while(root != null){
                stack.push(root);
                root = root.left;
            }   // root没有左孩子了，该轮到右孩子了
            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;
    }
}