树——二叉树的后序遍历(非递归)

来源:互联网 发布:江苏省建筑节能数据库 编辑:程序博客网 时间:2024/06/05 20:48

思路:

二叉树的后序遍历非递归方法与前序,中序不同,稍微麻烦一些。

       要保证根结点在左孩子和右孩子访问之后才能访问,因此对于任一结点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;    }}


0 0
原创粉丝点击