题目

从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。

 

例如: 给定二叉树: [3,9,20,null,null,15,7]

  3

 / \

9  20

    /  \

    15   7

返回其层次遍历结果:

[ [3], [9,20], [15,7] ]

来源:力扣(LeetCode)

链接:https://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof

解题思路

使用一个队列存储每一层的节点,每次循环都记录下当前队列的大小(n),n即为每层的节点数量,当队列不为空时, 每次出队n个节点同时保存节点的值到一个list,直到队列为空,最后返回结果

代码实现

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
class Solution {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>>  res =new ArrayList<>();
        if(root == null) return res;
        Queue<TreeNode> queue=new LinkedList<>();
        queue.offer(root);
        while(!queue.isEmpty()){
            List<Integer> tmp=new ArrayList<>();
            int n=queue.size();
            for(int i=0;i<n;i++){
                TreeNode node=queue.poll();
                tmp.add(node.val);
                if(node.left !=null) queue.offer(node.left);
                if(node.right !=null) queue.offer(node.right);
            }
            res.add(tmp);

        }
        return res;
    }
}