Leetcode 437. 路径总和 III(Python3)

最新推荐文章于 2024-01-17 23:54:41 发布

Aries888

最新推荐文章于 2024-01-17 23:54:41 发布

阅读量191

点赞数

CC 4.0 BY-SA版权

分类专栏： python3 leetcode 文章标签： leetcode python3

本文链接：https://blog.youkuaiyun.com/qq_38575545/article/details/88699708

python3 同时被 2 个专栏收录

181 篇文章

订阅专栏

leetcode

157 篇文章

订阅专栏

博客围绕LeetCode 437题路径总和III展开，给定一个每个结点存放整数值的二叉树，需找出路径和等于给定数值的路径总数。路径无需从根节点开始或在叶子节点结束，但方向必须向下，还给出了二叉树节点数量和节点数值范围。

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

437. 路径总和 III

给定一个二叉树，它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。

示例：

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8

      10
     /  \
    5   -3
   / \    \
  3   2   11
 / \   \
3  -2   1

返回 3。和等于 8 的路径有:

1.  5 -> 3
2.  5 -> 2 -> 1
3.  -3 -> 11

class Solution:
    def pathSum(self, root, sum):
        """
        :type root: TreeNode
        :type sum: int
        :rtype: int
        """
        def dfs(root, target, lst):
            if root == None:
                return 0
            res = 0
            v = lst[-1] + root.val
            for x in lst:
                if v - x == target:
                    res += 1
            lst.append(v)
            res += dfs(root.left, target, lst)
            res += dfs(root.right, target, lst)
            lst.pop()
            return res

        return dfs(root, sum, [0])