题目:
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
--------------
示例:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
提示:
节点总数 <= 1000
--------------------
思考:广度优先遍历 借助队列特性实现
题目要求的二叉树的 从上至下 打印(即按层打印),又称为二叉树的 广度优先搜索(BFS)。
BFS 通常借助 队列 的先入先出特性来实现。!!!
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/f19c36b1897c6f99a3943e39c5ec904c.png)
算法流程:
1、特例处理: 当树的根节点为空,则直接返回空列表 [] ;
2、初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root] ;
3、BFS 循环: 当队列 queue 为空时跳出;
1、出队: 队首元素出队,记为 node;
2、打印: 将 node.val 添加至列表 tmp 尾部;
3、添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue ;
4、返回值: 返回打印结果列表 res 即可。
--------------
复杂度分析:
时间复杂度 O(N) : N 为二叉树的节点数量,即 BFS 需循环 N 次。
空间复杂度 O(N): 最差情况下,即当树为平衡二叉树时,
最多有 N/2个树节点同时在 queue 中,使用 O(N)大小的额外空间。
----------------------
class Solution {
public int[] levelOrder(TreeNode root) {
if (root == null) return new int[0];
Queue<TreeNode> queue = new LinkedList<>(){};
queue.add(root);
ArrayList<Integer> list = new ArrayList<>();
while (!queue.isEmpty()) {
TreeNode node = queue.poll();
list.add(node.val);
if (node.left != null) queue.add(node.left);
if (node.right != null) queue.add(node.right);
}
int[] res = new int[list.size()];
for (int i = 0; i < list.size(); i++) {
res[i] = list.get(i);
}
return res;
}
}
LC