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.

**难度**: Medium

**标签**: 树、 深度优先搜索、


# -*- coding: utf-8 -*-
# @Author  : LG

"""
执行用时:40 ms, 在所有 Python3 提交中击败了81.35% 的用户
内存消耗:13.8 MB, 在所有 Python3 提交中击败了5.27% 的用户

解题思路:
    递归,获得每条支线节点组成的数字,然后相加
"""
class Solution:
    def sumNumbers(self, root: TreeNode) -> int:
        record = []

        def find(root, current):
            current += str(root.val)
            if root.left == None and root.right == None:
                record.append(int(current))
                return
            else:
                if root.left:
                    find(root.left, current)
                if root.right:
                    find(root.right, current)

        if root:
            find(root, '')

        return sum(record)