leetcode之Sum Root to Leaf Numbers

howtogetout

于 2015-12-30 10:34:58 发布

阅读量306

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/howtogetout/article/details/50432933

本文介绍了一种解决树节点数值求和的方法，通过遍历二叉树的所有路径，并将每条路径上的节点值连接成一个整数，最后求得所有路径对应整数的总和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

跟path sum系列用同一种思路，将所有的跟到叶提取出来，再组合成为一个数字，进行相加。代码如下：

# 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
        """
        pathlist = []
        def showpath(list1, root):
            if not root:
                return []
            list2 = list1[::]
            if list2 == []:
                list2.append(root.val)
            else:
                list2.append(root.val)
            if not root.left and not root.right:
                pathlist.append(list2)
            else:
                if root.left:
                    list1 = list2
                    showpath(list1, root.left)
                if root.right:
                    list1 = list2
                    showpath(list1, root.right)
            return pathlist
        s =  showpath([], root)
        if s == []:
            return 0
        for i in range(len(s)):
            for j in range(len(s[i])):
                s[i][j] = str(s[i][j])
        for i in range(len(s)):
            s[i] = int(''.join(s[i]))
        return sum(s)