看一百遍美女,美女也不一定是你的。但你刷一百遍算法,知识就是你的了~~
谁能九层台,不用累土起!
题目
给定一棵二叉树,其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法,打印节点数值总和等于某个给定值的所有路径的数量。注意,路径不一定非得从二叉树的根节点或叶节点开始或结束,但是其方向必须向下(只能从父节点指向子节点方向)。
示例:
给定如下二叉树,以及目标和 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
}
如有任何问题或建议,欢迎留言讨论!