leetcode_437

这道题目很有意思,比之前几道路径和题目都要难.

算法是参考了大佬们的解法

给出代码,解释在注释里

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