剑指 Offer 32 - I. 从上到下打印二叉树

题目

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
从上到下打印出二叉树的每个节点,同一层的节点按照从左到右的顺序打印。

例如:
给定二叉树: [3,9,20,null,null,15,7],

3
/ \
9 20
/ \
15 7
返回:
[3,9,20,15,7]

提示:
节点总数 <= 1000

思路

就是数的广度遍历,但是这个使用int数据接收的,我们可以先用一个list接收,然后最后把list放入数组中,这里可以使用for循环遍历也可以stream流

实现

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
private static class Solution {
public int[] levelOrder(TreeNode root) {
if (root == null) {
return new int[0];
}
Deque<TreeNode> queue = new ArrayDeque<>();
List<Integer> result = new ArrayList<>();
queue.add(root);
while (!queue.isEmpty()) {
TreeNode temp = queue.poll();
result.add(temp.val);
if (temp.left != null) {
queue.add(temp.left);
}

if (temp.right != null) {
queue.add(temp.right);
}

}

return result.stream().mapToInt(Integer::intValue).toArray();
}
}