力扣热题100 - 二叉树：路径总和 3

题目描述：

题号：437

给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。

解题思路：

思路一：对每个节点都进行深度优先搜索

对于树中的每个节点，我们都可以将其视为根节点，并计算以该节点为根的子树中所有路径和等于targetSum的路径数量。

关键点：将每个节点都视为根节点的递归

时间复杂度：O(N ^ N)

空间复杂度：O(N) 

C++

// C++
/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
    int sumNum(TreeNode* root, long long int targetSum) {
        if(root == nullptr) {
            return 0;
        }
        int count = 0;
        if(root->val == targetSum) {
            count++;
        }

        count += sumNum(root->left, targetSum - root->val);
        count += sumNum(root->right, targetSum - root->val);
        return count;
    }
public:
    int pathSum(TreeNode* root, int targetSum) {
        if(root == nullptr) {
            return 0;
        }

        // 从每个节点开始搜索
        int answer = sumNum(root, targetSum);
        answer += pathSum(root->left, targetSum);
        answer += pathSum(root->right, targetSum);
        return answer;
    }
};

go

// go
/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func sumNum(root *TreeNode, targetSum int64) int {
    if root == nil {
        return 0
    }
    count := 0
    if int64(root.Val) == targetSum {
        count++
    }
    count += sumNum(root.Left, targetSum - int64(root.Val))
    count += sumNum(root.Right, targetSum - int64(root.Val))
    return count
}

func pathSum(root *TreeNode, targetSum int) int {
    if root == nil {
        return 0
    }
    answer := 0
    // 从每个节点开始搜索
    answer += sumNum(root, int64(targetSum))
    answer += pathSum(root.Left, targetSum)
    answer += pathSum(root.Right, targetSum)
    return answer
}