【leetcode】【145】Binary Tree Postorder 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?

二、问题分析

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)。

三、Java AC代码

public ArrayList<Integer> postorderTraversal(TreeNode root) {
    ArrayList<Integer> res = new ArrayList<Integer>();
    helper(root, res);
    return res;
}
private void helper(TreeNode root, ArrayList<Integer> res)
{
    if(root == null)
        return;
    helper(root.left,res);
    helper(root.right,res);
    res.add(root.val);
}

最好是画一个树，模拟几个节点的操作，会清晰很多。

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