leetcode [1161] 最大层内元素和

---

Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub

---

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

class Solution:
    def maxLevelSum(self, root: TreeNode) -> int:
        level = [root]
        res, cur = 1, 1
        res_sum = 0
        while level:
            cur_sum = sum([lev.val for lev in level])
            if cur_sum > res_sum:
                res_sum, res = cur_sum, cur
            newlevel = []
            for lev in level:
                if lev.left:
                    newlevel.append(lev.left)
                if lev.right:
                    newlevel.append(lev.right)
            level = newlevel
            cur += 1
        return res