LeetCode 637. 二叉树的层平均值

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

 

示例 1：

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

提示：

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

思路：

这一题，难度倒是不难，就是才接触LeetCode的OJ没几天，不太熟悉他的风格，搞得头有点大= =
问题出在全局变量上，因为后端的判题机制是多次运行main函数这一点不清楚，所以我只在开头初始化了一下，结果就导致我自己执行代码是对的，一送到后端判题机就错了，找了半天还不知道问题在哪(┙>∧<)┙へ┻┻
解决方法当然就是，在main函数里进行初始化就OK啦~
关于这题的解题方法，其实有很多，不同的遍历方法都可以，看个人喜好~我用滴是先序（代码少写起来快~）

代码：

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
int counts[1010]={0};
double value[1010]={0};
int max_depth=0;
void preorder(TreeNode * root,int depth)
{
    if(root==NULL)
        return;
    max_depth=depth>max_depth?depth:max_depth;
    counts[depth]++;
    value[depth]+=root->val;
    preorder(root->left,depth+1);
    preorder(root->right,depth+1);
}
class Solution {
public:
    vector<double> averageOfLevels(TreeNode* root) {
        vector<double> answer;
        memset(counts,0,sizeof(counts));
        memset(value,0,sizeof(value));
        max_depth=0;
        preorder(root,0);
        for(int i=0;i<=max_depth;++i)
        {
            answer.push_back(value[i]/counts[i]);
        }
        return answer;
    }
};