题解 | #牛群平均重量#

考察的知识点：二叉树的层次遍历；

解答方法分析：

1. 首先检查根节点是否为空，如果为空，直接返回空的结果向量。
2. 创建一个队列，并将根节点入队。
3. 进入一个循环，每次循环一层节点：获取当前层的节点个数，通过记录队列长度 len实现。使用一个变量 sum 来保存当前层节点值的和。依次从队列中弹出当前层的节点，将其值加到 sum 中，并将其左子节点和右子节点（如果存在）入队。
4. 计算当前层的平均值，即将 sum 除以 len，并将结果加入到结果向量中。
5. 循环结束后，返回结果向量作为函数的返回值。

所用编程语言：C++；

完整编程代码：↓

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类
     * @return double浮点型vector
     */
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double> res;
        if (root == nullptr) {
            return res;
        }

        queue<TreeNode*> queue;
        queue.push(root);
        while (!queue.empty()) {
            int len = queue.size();
            double sum = 0;
            for (int i = 0; i < len; i++) {
                TreeNode* temp = queue.front();
                queue.pop();
                sum += temp->val;
                if (temp->left != nullptr) {
                    queue.push(temp->left);
                }
                if (temp->right != nullptr) {
                    queue.push(temp->right);
                }
            }
            res.push_back(sum / len);
        }

        return res;
    }
};