leetcode [面试题04.12] 求和路径

---

Contact me:

Blog -> https://cugtyt.github.io/blog/index
Email -> cugtyt@qq.com
GitHub -> Cugtyt@GitHub

---

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

示例:

给定如下二叉树，以及目标和 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]

提示：

节点总数 <= 10000

注意判断负数存在的可能情况，在目标值为0的情况下依然可以有其他路径。

class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> int:
        if not root: return 0
        def core(root, target):
            if not root: return 0
            if target == root.val: return 1 + core(root.left, 0) + core(root.right, 0)
            return core(root.left, target - root.val) + core(root.right, target - root.val)
        return core(root, sum) + self.pathSum(root.left, sum) + self.pathSum(root.right, sum)