剑指 Offer 32 - I. 从上到下打印二叉树
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。
例如:
给定二叉树: [3,9,20,null,null,15,7]
,
3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]
提示:
节点总数 <= 1000
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
//二叉树的层次遍历都是循环,循环就是一层一层的遍历,递归就是添加当前层的节点值
// 先初始化存结果的列表list 和 包含根节点的队列 queue
// BFS 循环:当队列 queue 为空时跳出
// 出队: 队首元素出队,记为 node;
// 打印: 将 node.val 添加至列表尾部;
// 添加子节点: 若 node 的左(右)子节点不为空,则将左(右)子节点加入队列 queue
class Solution {
public int[] levelOrder(TreeNode root) {
Queue<TreeNode> que = new LinkedList<>();
if (root != null) {
que.add(root);
}
int[] temp = new int[1010];
int count = 0;//计算有多少个有效节点
while(!que.isEmpty()) {
TreeNode cur = que.poll();
temp[count++] = cur.val;
if (cur.left != null)
que.add(cur.left);
if (cur.right != null)
que.add(cur.right);
}
int[] result = new int[count];
for (int i = 0; i < count; i++) {
result[i] = temp[i];
}
return result;
}
}