Python刷leetcode113. 路径总和 II

最新推荐文章于 2020-03-28 23:07:04 发布

Paper_Yu

最新推荐文章于 2020-03-28 23:07:04 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签： Leetcode Python

本文链接：https://blog.youkuaiyun.com/weixin_40314737/article/details/80947860

Leetcode 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种算法，该算法能够找到给定二叉树中所有从根节点到叶子节点路径总和等于特定目标值的路径。通过递归方式遍历二叉树，并在满足条件时记录路径。

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

给定一个二叉树和一个目标和，找到所有从根节点到叶子节点路径总和等于给定目标和的路径。

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

示例:
给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

[
   [5,4,11,2],
   [5,8,4,5]
]

思路：跟求是否有路径总和等于target的问题112基本相同，只是在计算过程中保留遍历路径结点组成的列表，如果满足条件就加入最终结果中

# 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 pathSum(self, root, sum):
        """
        :type root: TreeNode
        :type sum: int
        :rtype: List[List[int]]
        """
        self.res = []
        self.total = sum
        def caculate(root,sum,templist):
            if not root:
                return
            if not root.left and not root.right:
                if root.val+sum == self.total:
                    self.res.append(templist+[root.val])
            if root.left:
                caculate(root.left,sum+root.val,templist+[root.val])
            if root.right:
                caculate(root.right,sum+root.val,templist+[root.val])
        caculate(root,0,[])
        return self.res