Leetcode 1161：最大内层元素和

题目描述

给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。

请你找出层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中 最小 的那个。

 

示例：

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

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/maximum-level-sum-of-a-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

 

解题思路

class Solution {
public:
    int maxLevelSum(TreeNode* root) {
        int lev = 1,lsum = 0,msum = INT_MIN,ml = 1;
        queue<TreeNode*> que;
        TreeNode *pre = root;
        que.push(root);
        while(!que.empty()){
            root = que.front();
            lsum += root->val;
            if(root->left!=NULL) que.push(root->left);
            if(root->right!=NULL) que.push(root->right);
            if(pre == root){
                pre = que.back();
                if(lsum > msum){
                    msum = lsum;
                    ml = lev;
                }
                lev++;
                lsum = 0;
            }
            que.pop();
        }
        return ml;
    }
};