题目
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
提示:
节点总数 <= 1000
解题思路
1.题目要求我们从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。这道题与【剑指 Offer 32 - I. 从上到下打印二叉树】思路完全一致,大家可以先去学习一下。
2.唯一不同的就是需要将每一层打印到一行,那么我们就新建一个(List<List<Integer>> res = new ArrayList<>();)动态数组去存储元素,在每次while循环开始时,我们都需要统计出队列中元素的个数,因为在结束了一次 while 循环后,队列中存储的元素是同一层的元素,所以我们需要统计出个数,并且新建一个动态数组 cur 将一层的元素存储进去,并且等 cur 数组存储结束后,再将 cur 数组存入我们最终的 res 数组中。直到 队列为空,打印也就结束了。
代码实现
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
if(root == null){
return new ArrayList<>();
}
Queue<TreeNode> queue = new LinkedList<>();
List<List<Integer>> res = new ArrayList<>();
queue.add(root);
while(!queue.isEmpty()){
int k = queue.size();
List<Integer> cur = new ArrayList<>();
for(int i = 0; i < k; i++){
TreeNode t = queue.poll();
cur.add(t.val);
if(t.left != null) queue.add(t.left);
if(t.right != null) queue.add(t.right);
}
res.add(cur);
}
return res;
}
}
测试结果