[路飞]_leetcode-面试题 04.12-求和路径

题目描述

[题目地址]

给定一棵二叉树，其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法，打印节点数值总和等于某个给定值的所有路径的数量。注意，路径不一定非得从二叉树的根节点或叶节点开始或结束，但是其方向必须向下(只能从父节点指向子节点方向)。

示例:
给定如下二叉树，以及目标和 sum = 22，

              5
             / \
            4   8
           /   / \
          11  13  4
         /  \    / \
        7    2  5   1

返回:

3
解释： 和为 22 的路径有：[5,4,11,2], [5,8,4,5], [4,11,7]

提示：

• 节点总数 <= 10000

解题思路

本题要求我们在二叉树中，找出路径上节点和值等于目标值的路径的数量。这里的路径的起点可以是任何一个节点，终点也可以是任何一个节点，但是路径必须是从上到下的。
所以我们可以以任何一个节点为起点，向下查找路径，并记录节点和值，当和值等于目标值的时候，就说明我们找到了一个符合条件的路径，记录数量。这里递归函数的退出条件就是当我们找到一个空节点为止。

当我们处理完了这样的一个过程，就找到了所有符合条件的路径，也就得到了符合条件的路径的数量。

代码实现

var pathSum = function (root, sum) {
    const dfs = (root, sum) => {
        if (root == null) return 0;
        let val = sum - root.val;
        console.log(val)
        return (root.val == sum) + dfs(root.left, val) + dfs(root.right, val);
    }
    if (root == null) return 0;
    return dfs(root, sum) + pathSum(root.left,sum) + pathSum(root.right, sum);
};

至此我们就完成了leetcode-面试题 04.12-求和路径