層數最深葉子節點的和之超簡單

給你一棵二叉樹,請你返回層數最深的葉子節點的和。

示例:

輸入: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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章