[LeetCode] Binary Tree Postorder Traversal

给定一个二叉树，返回它的 后序 遍历。

示例:

输入: [1,null,2,3]
1

2
/
3

输出: [3,2,1]

引用自Code ganker：http://blog.csdn.net/linhuanmars/article/details/22009351

“接下来是迭代的做法，本质就是用一个栈来模拟递归的过程，但是相比于Binary Tree Inorder Traversal和Binary Tree Preorder Traversal，后序遍历的情况就复杂多了。我们需要维护当前遍历的cur指针和前一个遍历的pre指针来追溯当前的情况（注意这里是遍历的指针，并不是真正按后序访问顺序的结点）。具体分为几种情况：
（1）如果pre的左孩子或者右孩子是cur，那么说明遍历在往下走，按访问顺序继续，即如果有左孩子，则是左孩子进栈，否则如果有右孩子，则是右孩子进栈，如果左右孩子都没有，则说明该结点是叶子，可以直接访问并把结点出栈了。
（2）如果反过来，cur的左孩子是pre，则说明已经在回溯往上走了，但是我们知道后序遍历要左右孩子走完才可以访问自己，所以这里如果有右孩子还需要把右孩子进栈，否则说明已经到自己了，可以访问并且出栈了。
（3）如果cur的右孩子是pre，那么说明左右孩子都访问结束了，可以轮到自己了，访问并且出栈即可。
算法时间复杂度也是O(n)，空间复杂度是栈的大小O(logn)。实现的代码如下：

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        if (root == null)
            return res;
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.push(root);
        TreeNode pre = null;
        while (!stack.isEmpty()) {
            TreeNode cur = stack.peek();
            if (pre == null || cur == pre.left || cur == pre.right) {
                if (cur.left != null)
                    stack.push(cur.left);
                else if (cur.right != null)
                    stack.push(cur.right);
                else {
                    res.add(cur.val);
                    stack.pop();
                }                                    
            }
            else if (cur.left == pre) {
                if (cur.right != null)
                    stack.push(cur.right);
                else {
                    res.add(cur.val);
                    stack.pop();
                }        
            }
            else if (cur.right == pre) {
                res.add(cur.val);
                    stack.pop();
            }
            pre = cur;
        }
        return res;
    }
}

还有一种解法是先按“根右左”的顺序进行遍历，然后reverse输出就变成“左右根”了，代码根据先序遍历稍微改动即可：

class Solution {
    public List<Integer> postorderTraversal(TreeNode root) {
        List<Integer> res = new ArrayList<Integer>();
        if (root == null)
            return res;
        List<Integer> fres = new ArrayList<Integer>();
        Stack<TreeNode> stack = new Stack<TreeNode>();
        stack.push(root);
        while (!stack.isEmpty()) {
            TreeNode cur = stack.pop();
            fres.add(cur.val);
            if (cur.left != null)
                stack.push(cur.left);
            if (cur.right != null)
                stack.push(cur.right);
        }
        for (int i = fres.size()-1; i >= 0; i--)
            res.add(fres.get(i));
        return res;
    }
}