5052. 最大层内元素和

给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。

请你找出层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中 最小 的那个。

 

 参考

/**
 * 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:
    int record[10005];
    int n;
    int maxLevelSum(TreeNode* root) {
        memset(record,0,sizeof(record));
        
        n=0;  //n用来记录一共多少层
        dfs(root,1);
        int max=INT_MIN;
        int result=1;   
        for(int i=1;i<=n;i++){
            if(record[i]>max){
                max=record[i];
                result=i;
            }
        }
        return result;
    }
    void dfs(TreeNode* root,int cur){   //遍历root的左右
        if(root==NULL){
            return ;
        }
        if(n<cur){
            n=cur;
        }
        record[cur]+=root->val;
        dfs(root->left,cur+1);
        dfs(root->right,cur+1);
    }
};