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