JZ84 二叉树中和为某一值的路径(三)

JZ84 二叉树中和为某一值的路径(三)

给定一个二叉树root和一个整数值 sum ，求该树有多少路径的的节点值之和等于 sum 。

1.该题路径定义不需要从根节点开始，也不需要在叶子节点结束，但是一定是从父亲节点往下到孩子节点

2.总节点数目为n

3.保证最后返回的路径个数在整形范围内(即路径个数小于231-1)

数据范围:

0<=n<=10000<=n<=1000

-10^{9<=节点值<=10}9−109<=节点值<=109

假如二叉树root为{1,2,3,4,5,4,3,#,#,-1}，sum=6，那么总共如下所示，有3条路径符合要求

思路：此题代码结构基本为遍历二叉树的代码结构，因此使用dfs

但不同的是，当前节点可能被使用，也可能不被使用，因此代码中有dfs，和FindPath两个递归，dfs表示从当前节点一直加到它的子节点，FindPath也表示不考虑当前节点值的情况，具体看代码理解题意吧！

class Solution:
    def __init__(self):
        self.count = 0
    def FindPath(self, root: TreeNode, sum: int) -> int:
        if not root: return 0
        self.dfs(root, sum)           # 考虑当前节点
        self.FindPath(root.left, sum) # 不考虑当前节点
        self.FindPath(root.right, sum)
        return self.count
    def dfs(self, node, sum):   
        if not node: return
        sum-=node.val
        if sum==0:
            self.count+=1
        if node.left:
            self.dfs(node.left,sum)
        if node.right:
            self.dfs(node.right,sum)