0
点赞
收藏
分享

微信扫一扫

1161. 最大层内元素和 (二叉树的层序遍历)

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

相关推荐

0 条评论