面试:路径求和III (求和路径)
题目: 给定一个二叉树,它的每个结点都存放着一个整数值。
找出路径和等于给定数值的路径总数。
注意,路径不一定非得从二叉树的根节点或叶节点开始或结束,但是其方向必须向下(只能从父节点指向子节点方向)。
二叉树不超过1000个节点,且节点数值范围是 [-1000000,1000000] 的整数。
示例:
root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8
返回 3。和等于 8 的路径有:
- 5 -> 3
- 5 -> 2 -> 1
- -3 -> 11
- 重点在于需要计算所有子路径
class Solution:
def pathSum(self, root: TreeNode, sum: int) -> int:
if not root:
return 0
return self.path(root, sum) + self.pathSum(root.left, sum) + self.pathSum(root.right, sum)
def path(self, root, sum):
if not root:
return 0
res = 0
if root.val == sum:
res += 1
res += self.path(root.left, sum - root.val)
res += self.path(root.right, sum - root.val)
return res