求根节点到叶节点数字之和Python解法
生活随笔
收集整理的這篇文章主要介紹了
求根节点到叶节点数字之和Python解法
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
給你一個二叉樹的根節(jié)點 root ,樹中每個節(jié)點都存放有一個 0 到 9 之間的數(shù)字。
每條從根節(jié)點到葉節(jié)點的路徑都代表一個數(shù)字:
例如,從根節(jié)點到葉節(jié)點的路徑 1 -> 2 -> 3 表示數(shù)字 123 。
計算從根節(jié)點到葉節(jié)點生成的 所有數(shù)字之和 。
葉節(jié)點 是指沒有子節(jié)點的節(jié)點。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/sum-root-to-leaf-numbers
?
例:
輸入:root = [1,2,3]
輸出:25
解釋:
從根到葉子節(jié)點路徑 1->2 代表數(shù)字 12
從根到葉子節(jié)點路徑 1->3 代表數(shù)字 13
因此,數(shù)字總和 = 12 + 13 = 25
解析:
深度優(yōu)先搜索,每次向下傳遞時將已有值乘以10然后相加自身節(jié)點的值傳下去。
# Definition for a binary tree node. # class TreeNode(object): # def __init__(self, val=0, left=None, right=None): # self.val = val # self.left = left # self.right = right class Solution(object):def sumNumbers(self, root):""":type root: TreeNode:rtype: int"""def dfs(root, sum):if root is None: # 搜索到葉子節(jié)點后return 0sum = sum * 10 + root.val # 之前的值乘以10然后加上自身的值if root.left is None and root.right is None: # 沒有字節(jié)點直接返回return sumelse:return dfs(root.left, sum) + dfs(root.right, sum) # 有字節(jié)點就繼續(xù)向下傳遞return dfs(root, 0)總結
以上是生活随笔為你收集整理的求根节点到叶节点数字之和Python解法的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 小乔台词是什么(王者荣耀小乔)
- 下一篇: 正则表达式匹配Python解法