从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。 例如: 给定二叉树:
[3,9,20,null,null,15,7]
, 3 / \ 9 20 / \ 15 7 返回: [3,9,20,15,7]
null
,是则直接输出 []
list
用于存放二叉树层序遍历结果,同时用一个队列 queue
来存放二叉树层序遍历的节点;queue
不为空,则每次将队尾元素出队并存入 list
list
转换为数组返回即可,然后还有将列表转换为数组,时间复杂度为
,所以最终时间复杂度仍为
;
public int[] levelOrder(TreeNode root) {
// 队列为空,输出 []
if (root == null) {
return new int[0];
}
// 列表用于存放最终结果
ArrayList<Integer> list = new ArrayList<>();
// 队列用于存放树的顺序遍历
Queue<TreeNode> queue = new LinkedList<>();
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;
}