LeetCode二叉树 637. 二叉树的层平均值

给定一个非空二叉树的根节点 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]

class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
vector<double> ans;
queue<TreeNode*> q;

//TreeNode* node=nullptr;
q.push(root);

while(!q.empty()){
    double sum=0;
int count=q.size();
TreeNode* node=nullptr;
double n=0;
while(count--){
node=q.front();
q.pop();
    sum+=node->val;
    n++;
    if(node->left) q.push(node->left);
    if(node->right) q.push(node->right);
}
ans.push_back(sum/n);
}
return ans;

    }
};