思路:
二叉树的后序遍历非递归方法与前序,中序不同,稍微麻烦一些。
要保证根结点在左孩子和右孩子访问之后才能访问,因此对于任一结点cur,先将其入栈。如果cur不存在左孩子和右孩子,则可以直接访问它;或者cur存在左孩子或者右孩子,但是其左孩子和右孩子都已被访问过了,则同样可以直接访问该结点。若非上述两种情况,则将cur的右孩子和左孩子依次入栈,这样就保证了每次取栈顶元素的时候,左孩子在右孩子前面被访问,左孩子和右孩子都在根结点前面被访问。
代码如下:
import java.util.*;
public class Solution {
public ArrayList<Integer> postorderTraversal(TreeNode root) {
ArrayList<Integer> array=new ArrayList();
if(root == null)
return array;
Stack<TreeNode> stack=new Stack();
stack.push(root);
TreeNode pre=null;//定义pre
while(!stack.empty())
{
TreeNode cur=stack.peek();
if(cur.left == null&&cur.right == null||
(pre!=null&&(pre == cur.left||pre == cur.right)))//pre!=null不能少,否则root.left == null但是root.right!=null的情况会出错!
{
array.add(cur.val);
stack.pop();
pre=cur;
}else
{
if(cur.right!=null)
{
stack.push(cur.right);
}
if(cur.left!=null)
{
stack.push(cur.left);
}
}
}
return array;
}
}