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