JZ84 二叉树中和为某一值的路径(三)
给定一个二叉树root和一个整数值 sum ,求该树有多少路径的的节点值之和等于 sum 。
1.该题路径定义不需要从根节点开始,也不需要在叶子节点结束,但是一定是从父亲节点往下到孩子节点
2.总节点数目为n
3.保证最后返回的路径个数在整形范围内(即路径个数小于231-1)
数据范围:
0<=n<=10000<=n<=1000
-109<=节点值<=109−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)