LeetCode刷题笔录Binary Tree Post Order Traversal

最新推荐文章于 2019-08-04 18:14:36 发布

原创最新推荐文章于 2019-08-04 18:14:36 发布 · 766 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #traversal #遍历 #tree

LeetCode 同时被 2 个专栏收录

114 篇文章

订阅专栏

Tree

15 篇文章

订阅专栏

本文介绍了一种不使用递归实现二叉树后序遍历的方法，通过运用栈和前驱节点的概念，详细解释了每一步操作的目的及逻辑。此方法避免了递归带来的栈溢出风险。

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

Given a binary tree, return the postorder traversal of its nodes' values.

For example:
Given binary tree {1,#,2,3},

return [3,2,1].

Note: Recursive solution is trivial, could you do it iteratively?

Post order traversal的顺序是先访问左子树，再访问右子树，最后访问自身。这里可以用一个stack来存放待访问节点，并用一个prev指针来存储上一个访问的node，根据prev和curr的关系来决定如何前进。

每次peek一下stack的顶元素，作为curr。

情况分为：

1.prev是curr的parent

这时我们是在从上往下traverse。根据post order traversal的定义，如果curr.left != null，则将curr.left放到Stack里。如果curr.left = null,则将curr.right放进去。如果curr.left和curr.right都是null，说明curr是一个leaf node，访问之并从stack中pop出来。

2.prev是curr的left child

这时我们是在从左子树往上访问，按顺序应该将curr.right放入stack

3.prev是curr的right child

这时我们是从右子树往上访问，应该访问curr本身了。

注意只有当curr被访问时才会从stack里面将其pop出来。

public class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        if(root == null)
            return res;
        TreeNode prev = null;
        Stack<TreeNode> s = new Stack<TreeNode>();
        s.push(root);
        while(!s.isEmpty()){
            TreeNode curr = s.peek();
            //traversing down the tree
            if(prev == null || prev.left == curr || prev.right == curr){
                if(curr.left != null){
                    s.push(curr.left);
                }
                else if (curr.right != null){
                    s.push(curr.right);
                }
                else{
                    //curr is a leaf node, output it
                    res.add(curr.val);
                    s.pop();
                }
            }
            //traversing up the tree from left
            else if(curr.left == prev){
                if(curr.right != null){
                    s.push(curr.right);
                }
                else{
                    res.add(curr.val);
                    s.pop();
                }
            }
            //traversing up the tree from right
            else if(curr.right == prev){
                res.add(curr.val);
                s.pop();
            }
            prev = curr;
            
        }
        return res;
    }
    
    
}