题目:
给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层,而根节点的子节点位于第 2 层,依此类推。
请你找出层内元素之和 最大 的那几层(可能只有一层)的层号,并返回其中 最小 的那个。
1
/ \
7 0
/ \
7 -8
输入:[1,7,0,7,-8,null,null]
输出:2
解释:
第 1 层各元素之和为 1,
第 2 层各元素之和为 7 + 0 = 7,
第 3 层各元素之和为 7 + -8 = -1,
所以我们返回第 2 层的层号,它的层内元素之和最大。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/maximum-level-sum-of-a-binary-tree
思路:
一开始我想的是用递归算出每一层的和,在递归的同时把和最小的层数给算出来,显然我有点失败了
最后用的是递归算出和,然后暴力求最小值
上代码:
class Solution {
public:
int total[10001] = {0};
int maxnum=INT_MIN,t=0;
int maxLevelSum(TreeNode* root) {
bfs(root,0);
maxnum = root->val;
for(int i=0;i<10001;i++)
{
if(total[i]>maxnum)
{
maxnum = total[i];
t = i;}
}
return t+1;
}
void bfs(TreeNode *root, int depin)
{
total[depin] += root->val;
if(root->left!=NULL)
bfs(root->left,depin+1);
if(root->right!=NULL)
bfs(root->right,depin+1);
}
};