LeetCode刷题笔录Binary Tree Post Order Traversal

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

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

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