102. Binary Tree Level Order Traversal(二叉树的层次遍历)

最新推荐文章于 2019-05-28 09:58:21 发布

God_Mood

最新推荐文章于 2019-05-28 09:58:21 发布

阅读量226

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： Binary Tree Level Order Traver 二叉树的层次遍历队列递归

本文链接：https://blog.youkuaiyun.com/God_Mood/article/details/88639760

Leetcode 专栏收录该内容

227 篇文章

订阅专栏

博客围绕二叉树层次遍历问题展开，给出两种解法。一是用队列解决，关键在于区分节点所在层次，遍历一层时记录节点数并添加到列表；二是递归解法，官方题解虽简短，但本质仍运用队列思想。题目链接为https://leetcode.com/problems/binary-tree-level-order-traversal/ 。

题目链接：https://leetcode.com/problems/binary-tree-level-order-traversal/

方法一：

思路：用队列解决，难点在于如何区分第几层，

这里遍历到一层，就记录当前层有多少个节点，然后在list中加几个节点。

AC 1ms 77% Java：

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> ans=new ArrayList();
        if(root==null)
            return ans;
        Queue<TreeNode> queue=new LinkedList();
        queue.offer(root);
        int level=0;
        while(!queue.isEmpty()){
            ans.add(new ArrayList());
            int len=queue.size();
            for(int i=0;i<len;i++){
                TreeNode temp=queue.poll();
                ans.get(level).add(temp.val);
                if(temp.left!=null)
                    queue.offer(temp.left);
                if(temp.right!=null)
                    queue.offer(temp.right);
            }
            level++;
        }
        return ans;
    }
}

方法二：

递归，官方给的题解看起来简短，其实还是用的队列的思想。

代码如下：

class Solution {
    List<List<Integer>> levels = new ArrayList<List<Integer>>();

    public void helper(TreeNode node, int level) {
        // start the current level
        if (levels.size() == level)
            levels.add(new ArrayList<Integer>());

         // fulfil the current level
         levels.get(level).add(node.val);

         // process child nodes for the next level
         if (node.left != null)
            helper(node.left, level + 1);
         if (node.right != null)
            helper(node.right, level + 1);
    }
    
    public List<List<Integer>> levelOrder(TreeNode root) {
        if (root == null) return levels;
        helper(root, 0);
        return levels;
    }
}