二叉树后序遍历

二叉树后序遍历两种方法的对比

两种思想吧

一、root的左右节点先都放到stack中

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        LinkedList<Integer> res=new LinkedList<>();
        if(root==null)
            return res;
        Stack<TreeNode> stack=new Stack<>();
        stack.push(root);
        while(!stack.isEmpty()){
            TreeNode cur=stack.pop();
            //addFirst()是LinkedList中的方法，Link中是没有的，但是最后LinkedList可以直接向上转成List
            res.addFirst(cur.val);
            if(cur.left!=null){
                stack.push(cur.left);
            }
            if(cur.right!=null){
                stack.push(cur.right);
            }
        }
        return res;
    }
}

二、先保存右节点，然后再转向左节点

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        LinkedList<Integer> res=new LinkedList<>();
        Stack<TreeNode> stack=new Stack<>();
        while(!stack.isEmpty()||root!=null){
            //报错空指针
            //res.addFirst(root.val);
            //stack.push(root);
            if(root!=null){
//正确方法
                res.addFirst(root.val);
                stack.push(root);
                root=root.right;
            }else{
                root=stack.pop();
                root=root.left;
            }
        }
        return res;
    }
}