题目浅析

  • 想查看原题可以点击题目链接

  • 简单地说,就是给一个二叉树,求其中数值和最大的一层。

思路分享

代码解答(强烈建议自行解答后再看)

  • 参考题解
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
29
30
# 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:
layer = [root]
ans = 0
cur_index = 1
rec_s = -inf
while layer:
cur_s = 0
nxt_layer = []
for node in layer:
cur_s += node.val
if node.left:
nxt_layer.append(node.left)
if node.right:
nxt_layer.append(node.right)
if cur_s > rec_s:
ans = cur_index
rec_s = cur_s
# print(cur_s)

layer = nxt_layer
cur_index += 1

return ans