这道题目很有意思,比之前几道路径和题目都要难.
算法是参考了大佬们的解法
给出代码,解释在注释里
from collections import *
class Solution:
"""
注意路径不一定从根节点开始,也没有必要在叶子节点结束
"""
def __init__(self):
self.res = 0 # 保存最后结果
self.dic = defaultdict(int) # 对于每一个键值key,它所对应的值代表着从根节点到达当前搜索的节点,路径和为key的个数
self.dic[0] = 1
def pathSum(self, root: TreeNode, sum: int) -> int:
self.sum = sum
self.dfs(root, 0)
return self.res
def dfs(self, root: TreeNode, tempSum: int) -> None:
if root == None:
return
tempSum += root.val
self.res += self.dic[tempSum - self.sum] # 搜索路径上距离为tempSum-sum的个数,代表着从那些节点到当前节点路径和为sum
self.dic[tempSum] += 1 # 当前节点距离为tempSum
self.dfs(root.left, tempSum)
self.dfs(root.right, tempSum)
self.dic[tempSum] -= 1 # 搜索结束,减1