leetcode 637,Average of Levels in Binary Tree,难度,easy
0. 题干
给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。
示例 1:
输入:
输出:[3, 14.5, 11]
解释:
第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。
提示:
节点值的范围在32位有符号整数范围内。
1. 代码(含注释)
解题思路
对二叉树进行广度优先遍历,遍历到每一层时,计算均值,将结果添加到数组中。
思路还是BFS,代码没啥说的,比较简单。
class Solution {
public:
vector<double> averageOfLevels(TreeNode* root) {
queue<TreeNode*> que; //构造队列que,里面的元素都是TreeNode*类型的指针
que.push(root);//广度优先遍历,先将根节点添加到队列
vector<double> res; //存储结果
while (que.size()!= 0)
{
int n = que.size();
double sum=0; //sum用来存储每一层节点值的和
for ( int i=0;i<n;i++)
{
TreeNode* tmp=que.front(); //取出队列的第一个元素
que.pop();
sum=sum+tmp->val; //存储每一层的节点和
//如果该节点的左右孩子存在,就放到队列里面去
if(tmp->left)
que.push(tmp->left);
if(tmp->right)
que.push(tmp->right);
}
res.push_back(sum/n);
}
return res;
}
};