Leetcode刷题笔记 637. 二叉树的层平均值

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

示例1
输入：

    3
   / \
  9  20
    /  \
   15   7

输出：
[3, 14.5, 11]

解释：
第 0 层的平均值是 3 , 第1层是 14.5 , 第2层是 11 。因此返回 [3, 14.5, 11] 。

注意：
节点值的范围在32位有符号整数范围内。

代码

#include <stdio.h>
#include <vector>
#include <iostream>
#include <queue>
using namespace std;
// Definition for a binary tree node.
struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double> ans;
        queue<TreeNode*> q;
        q.push(root);
        while(!q.empty()){
            double sum = 0;
            int n = q.size();
            for(int i=0;i<n;i++){
                TreeNode *tmp = q.front();
                q.pop();
                sum += tmp->val;
                if(tmp->left)
                    q.push(tmp->left);
                if(tmp->right)
                    q.push(tmp->right);
            }
            ans.push_back(sum/n);
        }
        return ans;
    }
};
int main()
{
    TreeNode root(3);
    TreeNode l1(9);TreeNode r1(20);
    TreeNode l2(15);TreeNode r2(7);
    root.left = &l1;root.right = &r1;
    r1.left = &l2;r1.right=&r2;
    Solution s;
    vector<double> ans = s.averageOfLevels(&root);
    for(double x:ans){
        cout<<x<<endl;
    }
    return 0;
}

今天也是爱zz的一天哦！