[路飞]_求和路径

求和路径

题目

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

示例

给定如下二叉树，以及目标和 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]

题解

递归能解决二叉树80%的问题。

本题求所有路径的数量；且方向向下，难度降低了不少；

两次递归

• 第一次递归，将二叉树从根节点开始向下深度优先遍历；每个节点作为目标和的根节点；起始值设置为0；
• 从当前节点深度优先遍历，递归过程中增加节点值，直到等于目标和，结果+1；
• 返回结果

代码

var pathSum = function (root, sum) {
  if (root === null) return 0
  let result = 0
  dfs(root)
  function dfs(node) {
    if (node === null) return
    helper(node, 0)
    dfs(node.left)
    dfs(node.right)
  }
  return result
  // 得到数据
  function helper(node, total) {
    if (node === null) return
    if (total + node.val === sum) {
      result++
    }

    helper(node.right, total + node.val)
    helper(node.left, total + node.val)
  }
}