树 --- leedcode 437 统计路径和等于一个数的路径数量 （Easy）

题目

给定一个二叉树，它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

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

二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。

示例：

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

      10
     /  \
    5   -3
   / \    \
  3   2   11
 / \   \
3  -2   1

返回 3。和等于 8 的路径有:

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

解析

思路： 双重递归

• 既然路径不一定以 root 开头，也不一定以 leaf 结尾，但是必须连续。那么每个节点都可以作为根节点去dfs一次，只要存在满足条件的路径，就将结果加1.

js代码：

var pathSum = function(root, sum) {
    if(root == null) return 0;
    // 每一个节点都作为根节点遍历一次
    let res = dfs(root, sum)+ pathSum(root.left, sum) + pathSum(root.right, sum); 
    return res;
    
    // 每一个节点都要将其左右子树遍历完
    function dfs (root, num) {
        if(root == null) return 0;
        let ret = 0;
        if(root.val == num){
            ret ++;
        }  
        ret +=  dfs(root.left, num - root.val)+ dfs(root.right, num - root.val);
        return ret;
    }
};

java代码：

class Solution {
   public int pathSum(TreeNode root, int sum) {
    if (root == null) return 0;
    int ret = pathSumStartWithRoot(root, sum) + pathSum(root.left, sum) + pathSum(root.right, sum);
    return ret;
}

private int pathSumStartWithRoot(TreeNode root, int sum) {
    if (root == null) return 0;
    int ret = 0;
    if (root.val == sum) ret++;
    ret += pathSumStartWithRoot(root.left, sum - root.val) + pathSumStartWithRoot(root.right, sum - root.val);
    return ret;
}
}