二叉树后续遍历（递归与非递归）

原创于 2021-08-02 19:24:38 发布 · 117 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树 #java

本文介绍二叉树后序遍历的两种方法：递归与非递归。递归方式简洁直观；非递归则利用两个栈实现，先将根节点入栈，再分别入栈左右子节点，最后通过出栈顺序得到后序遍历结果。

后序遍历（递归）

  public void postorder(TreeNode root,List<Integer> list){
       if(root == null) return;                       
       postorder(root.left,list);
       postorder(root.right,list);
       list.add(root.val);
    }
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> list = new LinkedList();
        postorder(root,list);
        return list;
    }

非递归

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
            List<Integer> list = new ArrayList<>();
            //用两个栈来实现 s1是中转栈 最终用s2的出栈顺序才是后序遍历的结果
            Stack<TreeNode> s1 = new Stack<>();
            Stack<TreeNode> s2 = new Stack<>();
            if(root != null) s1.push(root);
            //s1的出栈顺序为根右左放入s2 所以s2的出栈顺序就是左右根 和后序遍历一样
            while(!s1.isEmpty()){
                root = s1.pop();
                s2.push(root);
                if(root.left != null){
                    s1.push(root.left);                   
                }
                if(root.right != null){
                    s1.push(root.right);
                }
            }
            while(!s2.isEmpty()){
                list.add(s2.pop().val);
            }
            return list;
    }
}