LeetCode437.Path Sum III

最新推荐文章于 2024-10-06 14:10:45 发布

Aopir

最新推荐文章于 2024-10-06 14:10:45 发布

阅读量191

点赞数

分类专栏： LeetCode 文章标签：算法 leetcode python

本文链接：https://blog.csdn.net/qq_41704019/article/details/107449756

版权

LeetCode 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

LeetCode112.Path Sum
LeetCode113.Path Sum II
在做本题前可以先参照之前两个关于路径的题目

You are given a binary tree in which each node contains an integer value.

Find the number of paths that sum to a given value.

The path does not need to start or end at the root or a leaf, but it must go downwards (traveling only from parent nodes to child nodes).

The tree has no more than 1,000 nodes and the values are in the range -1,000,000 to 1,000,000.

Example:

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

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

Return 3. The paths that sum to 8 are:

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

分析

此题情况比较复杂，首先不同于之前的112题和113题，这里的路径首尾未必一定要是根节点和叶节点，也就是会有四种可能：

起点为根节点，终点为叶节点；
起点为根节点，终点不是叶节点；
起点不是根节点，终点是叶节点；
起点不是根节点，终点也不是叶节点。

在起点不是根节点的情况下就必须考虑sum值的变化和不变化问题，也就是在当前点不包含在路径内时候，则当前sum不会变化；

在起点包含当前节点时候，则sum值会发生变化，即需要减去当前节点的值，sum=sum-root.val

当终点不是叶节点时候，也就是说某一个中间节点就满足了当前sum=root.val的条件，但是此时应该继续向下遍历，以确保后面存在节点和为0的情况不被忽略

显然，递归的方式可以解决此类问题。但是需要考虑到以上各种情况不被忽略，因此整体上根据sum分为两类，一类为变化的情况，一类为不变化的情况。

不变化的情况则仅需将root节点替换为root.left或root.right节点，直接递归原函数；
变化的情况，通过判断当前的节点值root.val是否与当前的sum值相等，如果相等则path_num数目变化，否则更新sum值为sum-root.val继续递归test函数进行比较。
为了保证所有递归下的path_num的变化都不会漏失，建立类下的全局变量self.path_num来进行路径数目表示并返回。

Code

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def __init__(self):
        self.path_num = 0
        
    def pathSum(self, root: TreeNode, sum: int) -> int:
        if root:
            self.test(root,sum)
            self.pathSum(root.left, sum)
            self.pathSum(root.right, sum)  
        return self.path_num
        
        
    def test(self, root: TreeNode, sum: int) -> int:
        if not root:
            return None
        else:
            if root.val == sum:
                self.path_num += 1
            self.test(root.left, sum-root.val)
            self.test(root.right, sum-root.val)