Binary Tree Postorder Traversal

標籤：java   leetcode   tree   

題目

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?

方法

/** * 要保證根結點在左孩子和右孩子訪問之後才能訪問，因此對於任一結點P，先將其入棧。 * 如果P不存在左孩子和右孩子，則可以直接存取它；或者P存在左孩子或者右孩子，但是其左孩子和右孩子都已被訪問過了，則同樣可以直接存取該結點。 * 若非上述兩種情況，則將P的右孩子和左孩子依次入棧， * 這樣就保證了每次取棧頂元素的時候，左孩子在右孩子前面被訪問，左孩子和右孩子都在根結點前面被訪問。 */private Stack<TreeNode> stack = new Stack<TreeNode>();    public ArrayList<Integer> postorderTraversal(TreeNode root) {ArrayList<Integer> al = new ArrayList<Integer>();if (root != null) {TreeNode cur = null;TreeNode pre = null;    stack.push(root);    while (stack.size() != 0) {    cur = stack.peek();    if ((cur.left == null && cur.right == null) || (pre != null && (pre == cur.left || pre == cur.right))) {    al.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 al;    }