「这是我参与2022首次更文挑战的第9天,活动详情查看:2022首次更文挑战」
给定一棵二叉树,其中每个节点都含有一个整数数值(该值或正或负)。设计一个算法,打印节点数值总和等于某个给定值的所有路径的数量。注意,路径不一定非得从二叉树的根节点或叶节点开始或结束,但是其方向必须向下(只能从父节点指向子节点方向)。
示例:
给定如下二叉树,以及目标和 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) {
// 获取路径 传入节点和目标值
function getPath(node,target){
// 如果当前节点为空,递归结束,返回 0
// 因为空节点肯定是找不到等于目标值的路径的
if(node===null) return 0;
// 初始化数量为0,并记录当前节点值
let num = 0,nodeVal = node.val
// 如果当前节点值等于目标值,数量+1
if(nodeVal === target) num++
// 获取后续的目标值
const val = target-nodeVal
// 递归处理左子树
num += getPath(node.left,val)
// 递归处理右子树
num += getPath(node.right,val)
// 如果左子树不为空并且左子树没有被作为起点进行过查找
// 以左子树为起点查找路径
if(node.left&&!node.left.tag){
// 给左子树打上标记,证明作为起点使用过
node.left.tag = true;
num += getPath(node.left,sum)
}
// 如果右子树不为空并且右子树没有被作为起点进行过查找
// 以右子树为起点查找路径
if(node.right&&!node.right.tag){
// 给右子树打上标记,证明作为起点使用过
node.right.tag = true;
num += getPath(node.right,sum)
}
// 返回符合条件的路径的数量
return num;
}
// 返回结果值
return getPath(root,sum)
};
这里要注意的是,因为递归过程中,有的节点会被经过多次,所以要记录当前节点是否被当做起点进行过查找,防止重复查找。
至此我们就完成了 leetcode-面试题 04.12-求和路径
如有任何问题或建议,欢迎留言讨论!👏🏻👏🏻👏🏻