剑指Offer JZ84 二叉树中和为某一值的路径(三)

描述

给定一个二叉树root和一个整数值 sum ，求该树有多少路径的的节点值之和等于 sum 。

1.该题路径定义不需要从根节点开始，也不需要在叶子节点结束，但是一定是从父亲节点往下到孩子节点

2.总节点数目为n

3.保证最后返回的路径个数在整形范围内(即路径个数小于231-1)

数据范围:

0<=n<=10000<=n<=1000

-10^9<=节点值<=10^9−109<=节点值<=109

假如二叉树root为{1,2,3,4,5,4,3,#,#,-1}，sum=6，那么总共如下所示，有3条路径符合要求

示例1

输入：{1,2,3,4,5,4,3,#,#,-1},6

返回值：3

说明：如图所示，有3条路径符合

示例2

输入：{0,1},1

返回值：2

示例3

输入：{1,#,2,#,3},3

返回值：2

题目分析

这道题跟判断从根节点到叶子节点是否存在一条和为sum的路径，思路很像，只是要注意：

1、它没有限定必须从根节点出发，叶子节点终止，也就是可以以任一父节点出发，任一子节点终止，但不能从子节点出发往父节点回溯，需要遍历树中的每一个节点

2、当满足和为sum之后，但还没到叶子节点的时候，仍然可以继续往下寻找

3、对于每个节点，我们都是做同样的判断，也就是把当前节点值考虑到路径和里面判断是否满足sum，是否有左右节点，有的话需要继续往下寻找，因此采用递归的思想。

class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @param sum int整型 
     * @return int整型
     */
    int FindPath(TreeNode* root, int sum) {
        // write code here
        int num = 0;//记录从当前节点出发满足条件的路径条数
        if(root==nullptr)
        {
            return 0;
        }
        findpath(root,sum,num);//寻找当前节点出发满足条件的路径
        if(root->left)
        {
            num += FindPath(root->left,sum);//加上从左节点出发满足条件的路径条数
        }
        if(root->right)
        {
            num += FindPath(root->right,sum);//加上从右节点出发满足条件的路径条数
        }
        return num;
    }
    void findpath(TreeNode* root,int sum,int& num)
    {
        if(root==nullptr)
        {
            return;
        }
        if(sum==root->val)//找到一条满足条件的路径
        {
            num += 1;
        }
        //不满足条件继续向下寻找，满足条件了还是可以向下寻找
        findpath(root->left,sum-root->val,num);
        findpath(root->right,sum-root->val,num);
    }
};