题目描述
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
算法流程:
题目要求的二叉树的 从上至下 打印(即按层打印),又称为二叉树的 广度优先搜索(BFS)。
BFS 通常借助 队列 的先入先出特性来实现。
1.特例处理: 当树的根节点为空,则直接返回空列表 [] ;
2.初始化: 打印结果列表 res = [] ,包含根节点的队列 queue = [root] ;
3.BFS 循环: 当队列 queue 为空时跳出;
1.出队: 队首元素出队,记为 node;
2.打印: 将 node.val 添加至列表 tmp 尾部;
3.添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue ;
4.返回值: 返回打印结果列表 res 即可。
代码:
class Solution {
public int[] levelOrder(TreeNode root) {
if(root == null) return new int[0];
Queue<TreeNode> queue = new LinkedList<>(){{ add(root); }};
ArrayList<Integer> ans = new ArrayList<>();
while(!queue.isEmpty()) {
TreeNode node = queue.poll(); //取队列的首节点
ans.add(node.val); //入队
if(node.left != null) queue.add(node.left); //判断左边子节点是否为空
if(node.right != null) queue.add(node.right); //判断右边子节点是否为空
}
int[] res = new int[ans.size()];
for(int i = 0; i < ans.size(); i++)
res[i] = ans.get(i);
return res;
}
}