题解 | #牛群平均重量#

大家好，我是开车的阿Q，自动驾驶的时代已经到来，没时间解释了，快和阿Q一起上车。作为自动驾驶系统工程师，必须要有最好的C++基础，让我们来一起刷题吧。

题目考察的知识点

这道题目主要考察了二叉树的层次遍历和计算平均值。

题目解答方法的文字分析

要解决这道题目，我们需要计算每层牛的平均重量。可以使用层次遍历的方式来解决，每层从左到右遍历，并计算每层的平均重量。

思路步骤如下：

1. 初始化一个空的结果数组result，用于存储每层牛的平均重量。
2. 初始化一个队列q，将根节点入队。
3. 开始层次遍历，当队列不为空时，进行以下操作： a. 获取当前队列的大小，表示当前层的节点数。 b. 初始化一个变量sum，用于记录当前层的节点重量之和。 c. 遍历当前层的所有节点，将节点的值加到sum中，并将其左右子节点入队。 d. 计算当前层的平均重量，将其添加到result中。
4. 返回result数组。

本题解析所用的编程语言

本题解析所用的编程语言是C++。

完整且正确的编程代码

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double> result; // 存储每层牛的平均重量的结果数组
        
        if (!root) {
            return result;
        }
        
        queue<TreeNode*> q; // 队列用于层次遍历
        q.push(root);
        
        while (!q.empty()) {
            int size = q.size(); // 当前层的节点数
            double sum = 0; // 当前层的节点重量之和
            
            for (int i = 0; i < size; ++i) {
                TreeNode* node = q.front(); // 当前节点
                q.pop();
                
                sum += node->val; // 将节点的值加到sum中
                
                if (node->left) {
                    q.push(node->left); // 将左子节点入队
                }
                if (node->right) {
                    q.push(node->right); // 将右子节点入队
                }
            }
            
            result.push_back(sum / size); // 计算当前层的平均重量并添加到result中
        }
        
        return result;
    }
};

您的关注、点赞、收藏就是我创作的动力，三连支持阿Q！