目录
题目:
示例:
分析:
代码:
题目:
示例:
分析:
这道题和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;
}
};