【路飞】算法与数据结构-面试题 04.12. 求和路径

123 阅读1分钟

不管全世界所有人怎么说,我都认为自己的感受才是正确的。无论别人怎么看,我绝不打乱自己的节奏。喜欢的事自然可以坚持,不喜欢的怎么也长久不了。

LeetCode:原题地址

题目要求

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

示例 1: 给定如下二叉树,以及目标和 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

思路

递归

function getPathSum(root ,sum){
    if(root==null)return 0
    let val = sum - root.val
    return (root.val==sum) + getPathSum(root.left,val)+getPathSum(root.right,val)
}
var pathSum = function(root, sum) {
    if(root==null){return 0}
    // 分选这当前节点,和不选的情况
    return getPathSum(root,sum)+pathSum(root.left,sum)+pathSum(root.right,sum)
};