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