从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
解题思路:
- 从上到下打印二叉树,又称为二叉树的
广度优先搜索
(BFS) - BFS通常借助队列的先入先出特性来实现
算法流程:
- 特例处理: 当树的根节点为空,则直接返回空列表 [] ;
- 初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root] ;
- BFS 循环: 当队列 queue 为空时跳出;
- 出队: 对手元素出队,标记为
node
- 打印: 将
node.val
加入list 尾部 - 入队: 若
node
的左(右)子节点不为空,则将子节点依次加入到queue
队列中
- 出队: 对手元素出队,标记为
- 返回值: 返回打印列表
res
即可
代码:
class Solution {
public int[] levelOrder(TreeNode root) {
if(root == null) return new int[0];
Deque<TreeNode> queue = new LinkedList<>(); //存储树节点的队列
List<Integer> list = new ArrayList<>(); //存储数字的集合
queue.add(root);
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;
}
}