• 博客园logo
  • 会员
  • 众包
  • 新闻
  • 博问
  • 闪存
  • 赞助商
  • HarmonyOS
  • Chat2DB
    • 搜索
      所有博客
    • 搜索
      当前博客
  • 写随笔 我的博客 短消息 简洁模式
    用户头像
    我的博客 我的园子 账号设置 会员中心 简洁模式 ... 退出登录
    注册 登录

cynchanpin

  • 博客园
  • 联系
  • 订阅
  • 管理

View Post

二叉树遍历的非递归

前序遍历的非递归:1、在入栈时增加结果集,不停的取左子树入栈。直到为空。2、假设栈非空,pop栈顶结点。取其右子树作为当前结点,继续第一步。直到栈为空

中序遍历的非递归:1、在入栈时,不停的取左子树入栈,直到为空。2、假设栈非空,pop栈顶结点,增加结点集,取其右子树作为当前结点。继续第一步。直到栈为空

后序遍历的非递归:1、在遍历结点时,总是先将右子树结点入栈,再将左子树结点入栈。

2、假设左子树结点和右子树结点为空或者右子树结点已经訪问过,弹出栈顶元素,标记已訪问结点。增加结果集。直到栈为空。

详细代码例如以下:

class TreeNode
{
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

class Solution
{
    public List<Integer> preorderTraversal(TreeNode root)
    {
        List<Integer> ret = new ArrayList<Integer>();

        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode cur = root;

        while(cur != null || !stack.empty())
        {
            while (cur != null)
            {
                ret.add(cur.val);
                stack.push(cur);
                cur = cur.left;
            }

            if (!stack.empty())
            {
                TreeNode tmp = stack.pop();
                cur = tmp.right;
            }
        }

        return ret;
    }


    public List<Integer> inorderTraversal(TreeNode root)
    {
        List<Integer> ret = new ArrayList<Integer>();

        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode cur = root;

        while(cur != null || !stack.empty())
        {
            while (cur != null)
            {
                stack.push(cur);
                cur = cur.left;
            }

            if (!stack.empty())
            {
                TreeNode tmp = stack.pop();
                ret.add(tmp.val);
                cur = tmp.right;
            }
        }

        return ret;
    }

    public List<Integer> postorderTraversal(TreeNode root)
    {
        List<Integer> ret = new ArrayList<Integer>();

        Stack<TreeNode> stack = new Stack<TreeNode>();
        TreeNode cur = root, pre = null;

        if (null == cur) return null;

        stack.push(cur);
        while (!stack.empty())
        {
            TreeNode tmp = stack.peek();

            if ((tmp.left == null && tmp.right == null) || (pre != null && tmp.right == pre))
            {
                ret.add(tmp.val);
                stack.pop();
                pre = tmp;
            }
            else
            {
                if (tmp.right != null) stack.push(tmp.right);
                if (tmp.left != null) stack.push(tmp.left);
            }
        }

        return ret;
    }
}


posted on 2017-08-20 08:39  cynchanpin  阅读(223)  评论(0)    收藏  举报

刷新页面返回顶部
 
博客园  ©  2004-2025
浙公网安备 33010602011771号 浙ICP备2021040463号-3