给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层,而根节点的子节点位于第 2 层,依此类推。
请返回层内元素之和 最大 的那几层(可能只有一层)的层号,并返回其中 最小 的那个。
层序遍历:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def maxLevelSum(self, root: Optional[TreeNode]) -> int:
queue = collections.deque()
queue.append(root)
level = 0
bigest_sum = root.val
bigest_level = 1
while queue:
level += 1
level_sum = 0
length = len(queue)
for i in range(length):
rt = queue.popleft()
if rt.left: queue.append(rt.left)
if rt.right: queue.append(rt.right)
level_sum += rt.val
if level_sum > bigest_sum:
bigest_sum = level_sum
bigest_level = level
return bigest_level