题目:
给你一个二叉树的根节点 root
。设根节点位于二叉树的第 1
层,而根节点的子节点位于第 2
层,依此类推。
请你找出层内元素之和 最大 的那几层(可能只有一层)的层号,并返回其中 最小 的那个。
示例:
输入:[1,7,0,7,-8,null,null]
输出:2
解释:
第 1 层各元素之和为 1,
第 2 层各元素之和为 7 + 0 = 7,
第 3 层各元素之和为 7 + -8 = -1,
所以我们返回第 2 层的层号,它的层内元素之和最大。
AC代码:
int maxLevelSum(struct TreeNode* root){
struct TreeNode* queue[10010] = {NULL};
int front = 0;
int rear = 0;
queue[rear++] = root;
int max = -10000010;
int max_floor = 1;
int count_floor = 1;
int current_floor = 0;
while(front != rear){
++current_floor;
int temp_floor = count_floor;
count_floor = 0;
int sum = 0;
for(int i=0;i!=temp_floor;++i){
sum += queue[front]->val;
if(queue[front]->left){
queue[rear++] = queue[front]->left;
++count_floor;
}
if(queue[front]->right){
queue[rear++] = queue[front]->right;
++count_floor;
}
free(queue[front]);
queue[front] = NULL;
++front;
}
if(sum > max){
max = sum;
max_floor = current_floor;
}
}
return max_floor;
}