【LeetCode75】第四十题 最大层内元素和

目录

题目：

示例：

分析：

代码：

---

题目：

示例：

分析：

这道题和LeetCode75的上一题大同小异，都是要我们对二叉树进行层序遍历。

那具体如何层序遍历我再上一题也详细介绍过了，没看过或是不懂怎么层序遍历的小伙伴可以点开我的主页找一下LeetCode75的第三十九题。

那这道题我们可以对层序遍历做一些小改变，因为我们最终要的是每层元素的总和，所以在层序遍历的时候我们不需要把每个元素都存起来，我们可以直接加到对应层数的元素总和里。

最后再比较一下每层的元素总和大小，把最大的那一层的层数返回出去就可以了。

代码：

class Solution {
public:
    vector<int>temp;
    void digui(TreeNode* root,int deep){
        if(root==nullptr) return;
        if(deep>=temp.size()) temp.push_back(root->val);
        else temp[deep]+=root->val; 
        digui(root->left,deep+1);
        digui(root->right,deep+1);
    }
    int maxLevelSum(TreeNode* root) {
        int res=1;
        int max=root->val;
        digui(root,0);
        for(int i=1;i<temp.size();i++){
            if(temp[i]>max){
                max=temp[i];
                res=i+1;
            }
        }
        return res;
    }
};