题目描述
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
解题思路
queue只保存每一层的树的节点,所以循环遍历queue的大小就可以依次输出每层的节点了.
算法流程:
- 特例处理: 当根节点为空,则返回空列表 [] ;
- 初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root] ;
- BFS 循环: 当队列 queue 为空时跳出;
- 新建一个临时列表 tmp ,用于存储当前层打印结果;
- 当前层打印循环: 循环次数为当前层节点数(即队列 queue 长度);
- 出队: 队首元素出队,记为 node;
- 打印: 将 node.val 添加至 tmp 尾部;
- 添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue ;
- 将当前层结果 tmp 添加入 res 。
- 返回值: 返回打印结果列表 res 即可。
代码
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
if(root != null) queue.add(root);
List<List<Integer>> res = new ArrayList<>();
while(!queue.isEmpty()){
List<Integer> temp = new ArrayList<>(); //注意每遍历一层树节点,就要清空temp
//queue只保存每一层的树的节点,所以循环遍历queue的大小就可以依次输出每层的节点了
for(int i = queue.size(); i > 0; i--){ // 因为里面的queue在不断减小,所以不能用for(int i = 0; i < queue.size(); i++),不然就是两头一起减了,而int i = queue.size()只会被赋值一次,后面queue的大小不会影响它
TreeNode head = queue.poll();
temp.add(head.val); //注意temp应存储整数,而不是TreedNode
if(head.left != null) queue.add(head.left);
if(head.right != null) queue.add(head.right);
}
res.add(temp);
}
return res;
}
}
复杂度分析
- 时间复杂度 O(N) : N 为二叉树的节点数量,即 BFS 需循环 NN 次。
- 空间复杂度 O(N) : 最差情况下,即当树为平衡二叉树时,最多有 N/2 个树节点同时在 queue 中,使用 O(N) 大小的额外空间。