当前位置：网站首页>1161. 最大层内元素和（二叉树的层序遍历）

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

2022-07-31 10:14:00 【51CTO】

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

     1.
2.
3.
4.
5.
6.
7.
8.
9.
10.
11.
12.
13.
14.
15.
16.
17.
18.
19.
20.
21.
22.
23.
24.
25.
26.
27.
28.