NB33 牛群平均重量

描述

在一个牧场中，有很多牛。为了方便管理，牧场主将牛的重量排列成一棵二叉树。现在牧场主想知道每层牛的平均重量。请按照从上到下的顺序，返回每层牛的平均重量。

示例1

输入：

{1, 10, 20, #, #, 16, 2}

返回值：

[1.00000,15.00000,9.00000]

/**
 * struct TreeNode {
 *  int val;
 *  struct TreeNode *left;
 *  struct TreeNode *right;
 *  TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 * };
 */
#include <queue>
#include <vector>
class Solution {
  public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     *
     * @param root TreeNode类
     * @return double浮点型vector
     */
    vector<double> averageOfLevels(TreeNode* root) {
        // write code here
        vector<double> weight;
        if (!root) return weight;
        queue<TreeNode*>q;
        q.push(root);
        while (!q.empty()) {
            int num = q.size();
            double all = 0;
            for (int i = 0; i < num; i++) {
                TreeNode* t = q.front();
                q.pop();
                all += t->val;
                if (t->left) q.push(t->left);
                if (t->right) q.push(t->right);
            }
            weight.push_back(all/num);
        }
        return weight;
    }
};