题目
给定一个非空二叉树的根节点 root ,
以数组的形式返回每一层节点的平均值。与实际答案相差 10-5
以内的答案可以被接受。
示例 1：
img
1
2
3
4
| 输入：root = [3,9,20,null,null,15,7]
输出：[3.00000,14.50000,11.00000]
解释：第 0 层的平均值为 3,第 1 层的平均值为 14.5,第 2 层的平均值为 11 。
因此返回 [3, 14.5, 11] 。
|
示例 2:
img
1
2
| 输入：root = [3,9,20,15,7]
输出：[3.00000,14.50000,11.00000]
|
提示：
- 树中节点数量在
[1, 104] 范围内
-231 <= Node.val <= 231 - 1
题解
这一题没什么说法，也是简单题，和199是很像的，不多说了。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
| class Solution {
public List<Double> averageOfLevels(TreeNode root) {
Queue<TreeNode> queue = new LinkedList<>();
ArrayList<Double> arrayList = new ArrayList<>();
if (root != null) {
queue.add(root);
}
while (!queue.isEmpty()) {
double average = 0.0;
int size = queue.size();
for (int i = 0; i < size; i++) {
TreeNode node = queue.poll();
average += node.val;
if (node.left != null) {
queue.add(node.left);
}
if (node.right != null) {
queue.add(node.right);
}
}
arrayList.add(average / size);
}
return arrayList;
}
}
|