Leetcode 1161.最大层内元素和
1 题目描述(Leetcode题目链接)
给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层,而根节点的子节点位于第 2 层,依此类推。
请你找出层内元素之和 最大 的那几层(可能只有一层)的层号,并返回其中 最小 的那个。
输入:[1,7,0,7,-8,null,null]
输出:2
解释:
第 1 层各元素之和为 1,
第 2 层各元素之和为 7 + 0 = 7,
第 3 层各元素之和为 7 + -8 = -1,
所以我们返回第 2 层的层号,它的层内元素之和最大。
提示:
- 树中的节点数介于 1 和 1 0 4 10^4 104 之间
- − 1 0 5 < = n o d e . v a l < = 1 0 5 -10^5 <= node.val <= 10^5 −105<=node.val<=105
2 题解
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def maxLevelSum(self, root: TreeNode) -> int:
queue = collections.deque([root])
retv_level, curr_level = 0, 0
max_value = float("-inf")
while queue:
curr_level += 1
length = len(queue)
curr_sum = 0
while length > 0:
node = queue.popleft()
curr_sum += node.val
if node.left:
queue.append(node.left)
if node.right:
queue.append(node.right)
length -= 1
if curr_sum > max_value:
max_value = curr_sum
retv_level = curr_level
return retv_level