LeetCode 热题 HOT 100 第八十二天 437. 路径总和 III 用python3求解

最新推荐文章于 2024-04-13 21:14:27 发布

Triangulum

最新推荐文章于 2024-04-13 21:14:27 发布

阅读量230

点赞数

分类专栏： LeetCode 热题 HOT 文章标签： leetcode 算法力扣数据结构深度优先遍历

本文链接：https://blog.csdn.net/weixin_40634691/article/details/126494977

版权

LeetCode 热题 HOT 专栏收录该内容

100 篇文章 3 订阅

订阅专栏

这篇博客介绍了如何解决一个二叉树问题，即寻找节点值之和等于给定目标值的路径数量。通过深度优先搜索（DFS）策略，遍历二叉树的每个节点，并递归计算左子树和右子树满足条件的路径数。在每个节点上，如果当前节点值等于目标值，则增加路径计数。最后，返回包括当前节点在内的路径总数。示例展示了两种不同的二叉树结构及其对应的目标和路径数量。

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

题目地址

给定一个二叉树的根节点root，和一个整数targetSum，求该二叉树里节点值之和等于targetSum的路径的数目。
路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

示例 1：
输入：root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出：3
解释：和等于 8 的路径有 3 条，如图所示。

示例 2：
输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：3

提示:
二叉树的节点个数的范围是 [0,1000]
-10^9 <= Node.val <= 10^9
-1000 <= targetSum <= 1000
在这里插入图片描述

解法：DFS（深度优先搜索遍历）
参考官方题解：指路

官方题解的思路只讲了rootSum这个函数。我们需要自行注意pathSum这个函数。也就是重点注意理解以下两行代码：

ret += self.pathSum(root.left, targetSum)
ret += self.pathSum(root.right, targetSum)

代码实现：

class Solution:
    def pathSum(self, root: TreeNode, targetSum: int) -> int:
        def rootSum(root, targetSum):
            if root is None:
                return 0

            ret = 0
            if root.val == targetSum:
                ret += 1

            ret += rootSum(root.left, targetSum - root.val)
            ret += rootSum(root.right, targetSum - root.val)
            return ret
        
        if root is None:
            return 0
            
        ret = rootSum(root, targetSum) #以root为根节点进行计算,也就是在这个函数里面不管怎样计算sum,都是会带上root的值
        ret += self.pathSum(root.left, targetSum) #注意这里是pathSum函数,相当于是另一棵树了
        ret += self.pathSum(root.right, targetSum)
        return ret