Leetcode:1161. 最大层内元素和

题目:

给你一个二叉树的根节点 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;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值