当前位置 : 主页 > 编程语言 > python >

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

来源:互联网 收集:自由互联 发布时间:2022-08-10
给你一个二叉树的根节点 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
上一篇:python Spyder debug 出现异常
下一篇:没有了
网友评论