码迷,mamicode.com
首页 > 其他好文 > 详细

[LeetCode]Binary Tree Inorder Traversal

时间:2015-05-16 16:34:34      阅读:100      评论:0      收藏:0      [点我收藏+]

标签:二叉树   leetcode   traversal   

题意:题意比较简单就是二叉树的中序遍历
思路:1. 递归是最简单的了
   2. 非递归的实现:用一个stack做存储结构
一直查找leftchild 知道没有下一个leftchild,访问该元素
如果该元素有right child 加入stack

从stack中取出一个元素 重复上述行为(这里需要加入set集合记录访问过得left的节点 否则会出现重复访问的问题)

代码:

    public List<Integer> inorderTraversal(TreeNode root) {
        List<Integer> rs = new LinkedList<Integer>();
        inorderTraversal(rs, root);
        return rs;
    }

    /**
     * recursive solution
     * @param result
     * @param root
     */
    public void inorderTraversal(List<Integer> result,TreeNode root) {
        if(root == null)return;

        if(root.left != null) inorderTraversal(result, root.left);
        result.add(root.val);
        if(root.right != null) inorderTraversal(result, root.right);
    }

    /**
     * 中序遍历的非递归版本,记录访问标识set
     * @param root
     * @return
     */

    public List<Integer> inorderTraversal2(TreeNode root) {
        List<Integer> rs = new LinkedList<Integer>();
        Stack stack = new Stack();
        if(root == null) return rs;
        stack.push(root);

        Set<TreeNode> set = new HashSet<TreeNode>();
        while (! stack.isEmpty()){
            TreeNode node = (TreeNode) stack.peek();
            while (node.left != null){
                if(!set.contains(node.left)) {
                    stack.push(node.left);
                    set.add(node.left);
                } else
                    break;
                node = node.left;
            }
            node = (TreeNode) stack.pop();
            rs.add(node.val);
            if(node.right !=null ) stack.push(node.right);

        }
        return rs;
    }


[LeetCode]Binary Tree Inorder Traversal

标签:二叉树   leetcode   traversal   

原文地址:http://blog.csdn.net/youmengjiuzhuiba/article/details/45768753

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!