LeetCode Sum Root to Leaf Numbers

LeetCode解题之Sum Root to Leaf Numbers

原题

一棵树的每个节点都是0-9中的某一个数字，现在把从根节点到某一个叶子节点之间所有节点的数字依次连接起来组成一个新的数字。要求所有从根节点到叶子节点组成的数字的和。

注意点：

• 无

例子:

输入:

    1
   / \   2   3

输出: 25

解题思路

如果在叶子节点下再添加一个节点，则表示的数字可以通过将之前的数字乘以10再加上添加节点的数字。通过这一递推关系，我们可以递归地计算每条路径上的表示的数字，代码中参数s表示到当前节点之前该路径上所组成的数字。

AC源码

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


class Solution(object):
    def sumNumbers(self, root):
        """ :type root: TreeNode :rtype: int """
        return self._sumNumbers(root, 0)

    def _sumNumbers(self, root, s):
        if root is None:
            return 0
        s = s * 10 + root.val
        return sum([self._sumNumbers(r, s) for r in (root.left, root.right)]) or s

if __name__ == "__main__":
    None

欢迎查看我的Github (https://github.com/gavinfish/LeetCode-Python) 来获得相关源码。