【leetcode】437 路径总和III

最新推荐文章于 2024-08-24 13:50:23 发布

开拓算法

最新推荐文章于 2024-08-24 13:50:23 发布

阅读量289

点赞数

分类专栏： leetcode 文章标签：算法数据结构

本文链接：https://blog.csdn.net/luzhizw/article/details/111243426

版权

leetcode 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

这篇博客主要讨论了一种算法问题，涉及计算以特定节点为起点，在二叉树中找到和为目标值的路径数量。通过深度优先搜索（DFS）策略，递归地计算以当前节点、左子节点和右子节点为起点的路径数，并将这些路径数相加。当节点值等于目标和时，路径数增加1；否则，检查左右子树是否存在路径和为剩余和的路径。Python代码展示了这一算法的实现。

摘要由CSDN通过智能技术生成

题目

在这里插入图片描述

分析

每个节点都有可能是路径的开始，所以每次先算以当前节点T为开始的符合条件的路径有几条。然后分别算以左右节点为开始的路径有几条。最后返回T的路径条数+左子树的路径条数+右子树的路径条数。
每次算路径和是否等于target时，用sum减去当前节点的数值，如果为0，则当前节点正好是一个路径的尾节点，当前存在1条，如果不为0，那么算左右子树有没有符合sum-当前节点数值的和的路径。

python代码

class Solution(object):

    def pathSum(self, root, sum):
        """
        :type root: TreeNode
        :type sum: int
        :rtype: int
        """
        if not root:
            return 0
        res = self.dfs(root,sum)
        left = self.pathSum(root.left,sum)
        right = self.pathSum(root.right,sum)
        return res+left+right

    def dfs(self,root,sum):
        if not root:
            return 0
        res = 0
        if root.val == sum:
            res = 1
        
        return res+self.dfs(root.left,sum-root.val)+self.dfs(root.right,sum-root.val)