[路飞][LeetCode]面试题 04.12. 求和路径

133 阅读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

解题思路

  • 使用二叉树前序遍历把所有节点都当作根节点
  • 每个节点都使用一次dfs函数
  • 统计左右各节点中满足条件的个数,加起来进行汇总

解题代码

var pathSum = function(root, sum) {
    if(root==null) return 0
    return dfs(root,sum)+ pathSum(root.left,sum)+ pathSum(root.right,sum)
};

function dfs(node,sum){
    if(node==null) return 0
    let count = sum ===node.val?1:0
    count += dfs(node.left,sum-node.val)
    count += dfs(node.right,sum-node.val)
    return count
}

如有任何问题或建议,欢迎留言讨论!