lintcode--二叉树的后序遍历

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

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出一棵二叉树 {1,#,2,3},

   1
    \
     2
    /
   3

返回 [3,2,1]

//Recursive递归
public ArrayList<Integer> postorderTraversal(TreeNode root) {
    ArrayList<Integer> result = new ArrayList<Integer>();
    if (root == null) return result;
    result.addAll(postorderTraversal(root.left));
    result.addAll(postorderTraversal(root.right));
    result.add(root.val);
    return result;   
}
/*
后序遍历递归定义：先左子树，后右子树，再根节点。


后序遍历的难点在于：
需要判断上次访问的节点是位于左子树，还是右子树。


若是位于左子树，则需跳过根节点，先进入右子树，再回头访问根节点；
若是位于右子树，则直接访问根节点。
*/
//Iterative
public ArrayList<Integer> postorderTraversal(TreeNode root) {
    ArrayList<Integer> result = new ArrayList<Integer>();
    Stack<TreeNode> stack = new Stack<TreeNode>();
    TreeNode prev = null; // 以前走过的节点
    TreeNode curr = root;//当前访问的结点
    if (root == null) return result;


    stack.push(root);
    
    while (!stack.empty()) {
        curr = stack.peek();
       //一个根节点被访问的前提是：无右子树或右子树已被访问过
        if (prev == null || prev.left == curr || prev.right == curr) { 
            // 遍历树
            if (curr.left != null) {
                stack.push(curr.left);
            } else if (curr.right != null) {
                stack.push(curr.right);
            }
            
        } else if (curr.left == prev) { // 从左边穿过树
            if (curr.right != null) {
                stack.push(curr.right);
            }
        } else { // traverse up the tree from the right
            result.add(curr.val);
            stack.pop();
        }
        prev = curr;
    }
    return result;
}