题目简介：
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
示例 1：
1
2
3
4
5
6
7
8
9
| 输入：
3
/ \
9 20
/ \
15 7
输出：[3, 14.5, 11]
解释：
第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。
|
提示：
思路：
广搜，每次出队前先记录当前队列的大小，即该层的结点数量。
tip:
代码如下：
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
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
|
class Solution {
public:
vector<double> averageOfLevels(TreeNode* root) {
vector<double> res;
queue<TreeNode*> qu;
qu.push(root);
while(!qu.empty()){
int size = qu.size();
long sum = 0;
for(int i = 0; i < size; i++){
TreeNode* top = qu.front();
sum += top -> val;
qu.pop();
if(top -> left)
qu.push(top -> left);
if(top -> right)
qu.push(top -> right);
}
res.push_back(sum * 1.0 / size);
}
return res;
}
};
|
