面试题 04.12. 求和路径

面试题 04.12. 求和路径

原始题目链接：https://leetcode-cn.com/problems/paths-with-sum-lcci/
给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

解题思路：

这道题使用双递归的方法，一般遇到题目中有说任意节点开始XXX或者所有XXX这样问题，可以考虑使用双递归解法。具体实现看代码及注释。

代码实现：

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

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)

    # 定义一个递归函数：计算以当前节点为起点，路径和是sum的个数
    def path(self, root, sum):
        if not root:
            return 0
        ans = 0
        if root.val == sum:
            ans += 1
        # 递归当前节点的左右子树，符合条件的ans自增1
        ans += self.path(root.left, sum - root.val)
        ans += self.path(root.right, sum - root.val)
        return ans

参考文献：
https://leetcode-cn.com/problems/paths-with-sum-lcci/solution/python3xian-ji-suan-bao-han-gen-jie-dian-de-qing-2/