描述
给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。
示例 1：
输入：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:
输入：root = [3,9,20,15,7]
输出：[3.00000,14.50000,11.00000]提示：
树中节点数量在 [1, 104] 范围内
-231 <= Node.val <= 231 - 1
来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/average-of-levels-in-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
题解
C++代码
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
class Solution {
public:
vector<double> averageOfLevels(TreeNode* root) {
queue<TreeNode*> que; // 定义一个队列，用于层序遍历
que.push(root); // 将根节点入队
vector<double> result; // 结果
while(!que.empty()){
double sum = 0; // 每一层的和
int n = que.size();
for(int i=0; i<n; i++){ // 依次遍历当前层的所有节点
TreeNode* node = que.front(); // 取出队首节点
que.pop(); // 将队首节点出队
sum += node->val;
if(node->left) // 如果左子节点不为空，将其入队
que.push(node->left);
if(node->right) // 如果右子节点不为空，将其入队
que.push(node->right);
}
result.push_back(sum / n); // 求每一层的平均值
}
return result;
}
};Python代码
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def averageOfLevels(self, root: Optional[TreeNode]) -> List[float]:
res = [] # 保存结果
q = [root] # 定义一个队列，用于层序遍历
while q: # 当队列非空时
n = len(q) # 当前层的节点个数
sum_ = 0
for i in range(n): # 依次遍历当前层的所有节点
node = q.pop(0)
sum_ += node.val # 对当前层的节点值求和
if node.left:
q.append(node.left) # 如果左子节点不为空，将其入队
if node.right:
q.append(node.right) # 如果右子节点不为空，将其入队
res.append(sum_ / n) # 对每一层的和求平均值
return res
