給你一棵二叉樹,請你返回層數最深的葉子節點的和。
示例:
輸入:root = [1,2,3,4,5,null,6,7,null,null,null,null,8]
輸出:15
提示:
樹中節點數目在 1 到 10^4 之間。
每個節點的值在 1 到 100 之間。
思路:
我們只需要搜索所有的葉子結點,並把相同層數的葉子節點值加在一起
最後輸出層數最大的值
數據:
用字典:鍵爲層數,值爲數值
設計:
參數傳遞:子結點,字典,層數
基例:如果是葉子結點則根據層數增加值,返回字典
其它操作:無
遞歸:深度優先,先左子節點,後右子節點
返回值:新字典
class Solution:
def deepestLeavesSum(self, root: TreeNode) -> int:
dt={}
def dg(root,dt,s):
if not root.left and not root.right:
if not dt.get(s):
dt[s]=0
dt[s]+=root.val
return dt
if root.left:
dt=dg(root.left,dt,s+1)
if root.right:
dt=dg(root.right,dt,s+1)
return dt
if root:
dt=dg(root,dt,0)
dm=list(dt.keys())
dm.sort()
return dt[dm[-1]]
else:
return 0