leetcode_129. 求根到叶子节点数字之和

目录

一、题目内容

二、解题思路

三、代码


一、题目内容

给定一个二叉树,它的每个结点都存放一个 0-9 的数字,每条从根到叶子节点的路径都代表一个数字。

例如,从根到叶子节点路径 1->2->3 代表数字 123。

计算从根到叶子节点生成的所有数字之和。

说明: 叶子节点是指没有子节点的节点。

示例 1:

输入: [1,2,3]
    1
   / \
  2   3
输出: 25
解释:
从根到叶子节点路径 1->2 代表数字 12.
从根到叶子节点路径 1->3 代表数字 13.
因此,数字总和 = 12 + 13 = 25.

示例 2:

输入: [4,9,0,5,1]
    4
   / \
  9   0
 / \
5   1
输出: 1026
解释:
从根到叶子节点路径 4->9->5 代表数字 495.
从根到叶子节点路径 4->9->1 代表数字 491.
从根到叶子节点路径 4->0 代表数字 40.
因此,数字总和 = 495 + 491 + 40 = 1026.

二、解题思路

每次就将之前的和乘以10加当前节点的值,若没有左右孩子,则说明到达叶子节点,即个位数,则得到从根节点到该叶子结点的和,返回。左 + 右 再返回。

三、代码

# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

    def __repr__(self):
        return str(self.val)

class Solution:
    def sumNumbers(self, root: TreeNode) -> int:

        def dfs(root, sum):
            if not root:
                return 0
            elif root.left is None and root.right is None:
                return 10 * sum + root.val
            else:
                left = dfs(root.left, 10 * sum + root.val)
                right = dfs(root.right, 10 * sum + root.val)
                return left + right

        return dfs(root, 0)

if __name__ == '__main__':

    a = TreeNode(4)
    a.left = TreeNode(9)
    a.right = TreeNode(0)
    a.left.left = TreeNode(5)
    a.left.right = TreeNode(1)
    # a = TreeNode(1)
    # a.left = TreeNode(0)
    # a = TreeNode(1)
    # a.left = TreeNode(5)
    # a.right = TreeNode(1)
    # a.right.right = TreeNode(6)

    s = Solution()
    ans = s.sumNumbers(a)
    print(ans)
悲恋花丶无心之人 CSDN认证博客专家 深度学习 神经网络 Pytorch
计算机视觉在读研究生,熟悉Pytorch,MXNet,TensorFlow,Keras等深度学习框架,主要涉及的领域有目标检测,语义分割,超分辨率重建,行人重识别等。
个人GitHub网址为:https://github.com/nickhuang1996
©️2020 CSDN 皮肤主题: Age of Ai 设计师:meimeiellie 返回首页
实付 9.90元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、C币套餐、付费专栏及课程。

余额充值