python【⭐️】：求和路径

给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。
示例:
给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1
返回:
3
解释：和为 22 的路径有：[5,4,11,2], [5,8,4,5], [4,11,7]

一、回溯

class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> int:
		def path(root,prefixsum,target,cursum):
			if root == None:
				return 0
			res = 0
			cursum += root.val
			#在当前节点前是否有节点的前缀和+target= cursum，即为满足条件
			res += prefixsum.get(cursum-target,0)
			prefixsum[cursum] = prefixsum.get(cursum-target,0)+1
			#找子节点是否有符合条件的
			res+=path(root.left, prefixsum,target,cursum)
			res+=path(root.right, prefixsum,target,cursum)
			#回到本层
			prefixsum[cursum] = prefixSumCount.get(currSum) - 1
			return res
		
	    prefixsum = {}
	    prefixsum['0']=1
	    return path(root,prefixsum,sum,0)
    

二、两层遍历
遍历每个节点，再遍历每个节点后是否有子节点到当前节点的路径和满足条件。

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
cnt=0
class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> int:
        def dbs(root,target):
            global cnt
            if root == None:
                return
            target -= root.val
            if target == 0:
                cnt += 1
            dbs(root.left,target)
            dbs(root.right,target)

        def dbs2(root, target):
            if root == None:
                return
            dbs(root, target)
            dbs2(root.left,target)
            dbs2(root.right,target)
        global cnt #声明全局变量
        dbs2(root, sum)
        return cnt

python使用全局变量需要修改时要先声明。

三、数组+递归
用数组存放当前节点到它之前每个节点的距离和。

class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> int:
        def f(r, s):
            if r:
                s = [i + r.val for i in s] + [r.val]
                return s.count(sum) + f(r.left, s) + f(r.right, s)
            return 0
        return f(root, [])