LeetCode 437. Path Sum III 递归中使用递归，分两类情况

437. Path Sum III

You are given a binary tree in which each node contains an integer value.

Find the number of paths that sum to a given value.

The path does not need to start or end at the root or a leaf, but it must go downwards (traveling only from parent nodes to child nodes).

The tree has no more than 1,000 nodes and the values are in the range -1,000,000 to 1,000,000.

Example:

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8
10
/ \
5 -3
/ \ \
3 2 11
/ \ \
3 -2 1

Return 3. The paths that sum to 8 are:

1. 5 -> 3
2. 5 -> 2 -> 1
3. -3 -> 11

题意

给定一个二叉树，每个结点包含一个值，找到路径的数量其和等于给定的sum值

注意

• 和112. Path Sum有些类似，但是有些区别
• 这个题的路径起始并不需要根节点到叶子结点，但是必须向下的

思路

在112.Path Sum中从上往下递归，是假定node在路径中，所以在这题中，应该分为两种情况。
1.当前的node在路径中
2.当前的node不在路径中

代码

class Solution {
public:
    // 在以root为根节点的二叉树中,寻找和为sum的路径,返回这样的路径个数
    int pathSum(TreeNode* root, int sum) {
        if(root == NULL)
        {
            return 0;
        }
        int res = 0;
        //情况1：查找包含当前结点，是否存在其左右子树中路径其和为sum
        res += findPath(root,sum);

        //情况2：当前结点不包含在sum中，
        res+=pathSum(root->left,sum);
        res+=pathSum(root->right,sum);

        return res;
    }    
private:
    // 在以node为根节点的二叉树中,寻找包含node的路径,和为sum
    // 返回这样的路径个数
    int findPath(TreeNode* root,int num)
    {
        if(root == NULL)
        {
            return 0;
        }
        int res = 0;
        if(root->val == num)  //当前结点的值等于num表示找到一条路径
        {
            res+=1;
        }
        res += findPath(root->left,num-root->val);
        res += findPath(root->right,num-root->val);

        return res;
    }
};