题目:
给定一个二叉树的根节点 root ,和一个整数 targetSum ,
求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
路径 不需要从根节点开始,也不需要在叶子节点结束,
但是路径方向必须是向下的(只能从父节点到子节点)。
![在这里插入图片描述](https://i-blog.csdnimg.cn/blog_migrate/3f5d80ca0271b49c955c510c4f8fba88.png)
输入:root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出:3
解释:和等于 8 的路径有 3 条,如图所示。
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:3
提示:
二叉树的节点个数的范围是 [0,1000]
-109 <= Node.val <= 109
-1000 <= targetSum <= 1000
----------------------------------
思考:
这道题用到了一个概念,叫前缀和。就是到达当前元素的路径上,之前所有元素的和。
前缀和怎么应用呢?
在同一个路径之下(可以理解成二叉树从root节点出发,到叶子节点的某一条路径),
如果两个数的前缀总和是相同的,那么这些节点之间的元素总和为零。
进一步扩展相同的想法,
如果前缀总和currSum,在节点A和节点B处相差target,则位于节点A和节点B之间的元素之和是target。
因为本题中的路径是一棵树,从根往任一节点的路径上(不走回头路),有且仅有一条路径,因为不存在环。
(如果存在环,前缀和就不能用了,需要改造算法)
抵达当前节点(即B节点)后,将前缀和累加,然后查找在前缀和上,
有没有前缀和currSum-target的节点(即A节点),
存在即表示从A到B有一条路径之和满足条件的情况。
结果加上满足前缀和currSum-target的节点的数量。然后递归进入左右子树。
左右子树遍历完成之后,回到当前层,需要把当前节点添加的前缀和去除。
避免回溯之后影响上一层。因为思想是前缀和,不属于前缀的,我们就要去掉它。
代码:
class Solution {
public int pathSum(TreeNode root, int targetSum) {
Map<Long, Integer> map = new HashMap<>();
map.put(0L, 1);
return backTracking(root, map, targetSum, 0);
}
public int backTracking(TreeNode node, Map<Long, Integer> map, int targetSum, long curSum) {
if (node == null) return 0;
int res = 0;
curSum += node.val;
res += map.getOrDefault(curSum - targetSum, 0);
map.put(curSum, map.getOrDefault(curSum, 0) + 1);
res += backTracking(node.left, map, targetSum, curSum);
res += backTracking(node.right, map, targetSum, curSum);
map.put(curSum, map.get(curSum) - 1);
return res;
}
}
LC