Likou-437 경로 합계 III

에서 참조 https://leetcode-cn.com/problems/path-sum-iii/solution/qian-zhui-he-di-gui-hui-su-by-shi-huo-de-xia-tian/
접두사와 -현재 요소의 경로를 참조之前所有元素的和

• 동일한 경로에서 숫자가 두 개인 경우이 前缀和是相同的노드 사이의 요소 합계는 0입니다.
• 그런 다음 동일한 경로에서 노드 A의 접두사 합이 대상별 노드 B의 접두사 합과 다르면 노드 A와 노드 B 사이에있는 요소의 합이 대상이됩니다. (현재 알려진 경로가 A-> C-> B 인 경우 A와 B의 접두사 합계가 대상과 다르면 C + B의 합계가 대상입니다)

# 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 pathSum(self, root: TreeNode, sum: int) -> int:
        num_dict = {
    
    }
        num_dict[0] = 1	# 初始化，前缀和为0的一条路径
        return self.recursionPathSum(root, num_dict, sum, 0)
    
    def recursionPathSum(self, node, num_dict, target, currSum):
        if (node == None):  return 0
        res = 0
        currSum += node.val	# 当前路径和
        '''
        	重点:
        	如果此前有和为currSum-target,而当前的和又为currSum,两者的差就肯定为target了
        	所以根据当前num_dict的值进行搜索，将结果加入res中
        '''
        res += num_dict.get(currSum-target, 0)
        num_dict[currSum] = num_dict.get(currSum, 0)+1

        res += self.recursionPathSum(node.left, num_dict, target, currSum)
        res += self.recursionPathSum(node.right, num_dict, target, currSum)

        num_dict[currSum] -= 1	# 回到本层，去掉本层的路径和
        return res