题目
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [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;
}
}
|