最大层内元素和----leetcode周赛150_1002

题目描述:

给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层,而根节点的子节点位于第 2 层,依此类推。
请你找出层内元素之和 最大 的那几层(可能只有一层)的层号,并返回其中 最小 的那个。

示例 :

最大层内元素和----leetcode周赛150_1002_第1张图片

输入:[1,7,0,7,-8,null,null]
输出:2
解释:
第 1 层各元素之和为 1,
第 2 层各元素之和为 7 + 0 = 7,
第 3 层各元素之和为 7 + -8 = -1,
所以我们返回第 2 层的层号,它的层内元素之和最大。

提示

  1. 树中的节点数介于 1 和 10^4 之间
  2. -10^5 <= node.val <= 10^5

思路:

  • 利用队列,对二叉树进行层序遍历,记录每层节点和,逐步更新ans为最大和所在层数
  • 可以利用变量,记录下一层节点的个数,每次利用for循环访问一层的节点,即可达到统计的目的

具体代码如下 :

class Solution {
public:
    int ans =  0;int anshe = -2147483648;//ans层数与ans层和
    int cenghe; int curceng;
    queue que;
    int maxLevelSum(TreeNode* root) {
        if(!root)return 0;
        int len =1;int next = 0;curceng = 0;//当前层个数,下一层节点个数,当前在第几层。
        que.push(root);
        while(que.size()){
            cenghe = 0;//当前层的和
            curceng ++;//当前在第几层
            next = 0;//下一层个数
            for(int i  = 0 ; i < len;i++){
                TreeNode* t = que.front();que.pop();
                cenghe += t->val;
                if(t->left){que.push(t->left);next++;}
                if(t->right){que.push(t->right);next++;}
            }
            len = next;
            if(cenghe > anshe){
                ans = curceng;
                anshe = cenghe;
            }
        }
        return ans;
    }
};

你可能感兴趣的:(最大层内元素和----leetcode周赛150_1002)