leetcode算法（637.二叉树的层平均值）

本题就是层序遍历的时候把一层求个总和再取一个均值。

class Solution {
public:
    // 计算二叉树每一层节点值的平均值
    vector<double> averageOfLevels(TreeNode* root) {
        // 创建队列用于BFS层序遍历
        queue<TreeNode*> que;
        
        // 如果根节点不为空，将其加入队列
        if (root != NULL) que.push(root);
        
        // 存储结果的向量，将保存每一层的平均值
        vector<double> result;
        
        // 当队列不为空时，继续遍历
        while (!que.empty()) {
            // 获取当前层的节点数量
            int size = que.size();
            
            // 初始化当前层节点值的总和
            double sum = 0; // 使用double防止整数溢出和保留小数
            
            // 遍历当前层的所有节点
            for (int i = 0; i < size; 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 / size); // sum为double，size为int，结果为double
        }
        
        // 返回所有层平均值的向量
        return result;
    }
};