利用队列求二叉树每层的平均值

leetcode 637. 二叉树的层平均值

题目链接

给定一个非空二叉树, 返回一个由每层节点平均值组成的数组。

示例 1：
输入：
3
/
9 20
/
15 7
输出：[3, 14.5, 11]
解释：
第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。

  struct TreeNode {
      int val;
      TreeNode *left;
      TreeNode *right;
      TreeNode(int x) : val(x), left(NULL), right(NULL) {}
  };
 
 vector<double> averageOfLevels(TreeNode* root) {
        vector<double> result;
        if(root == nullptr)
            return result;
        queue<TreeNode*> que;
        que.push(root);
        while(!que.empty()){
            int size = que.size();
            double sum = 0.0;
            for(int i = 0;i < size; ++i){
                TreeNode* node = que.front();
                sum += node->val;
                que.pop();
                if(node->left)
                    que.push(node->left);
                if(node->right)
                    que.push(node->right);
            }
            result.push_back(sum / size);
        }
        return result;
    }