1.题目描述
给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
示例 1:
输入:root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出:3
解释:和等于 8 的路径有 3 条,如图所示。
示例 2:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:3
2.解题思路与代码
2.1 解题思路
这道题目要求每个节点开始的路径和等于 targetSum 的路径数目,我们可以通过遍历二叉树,然后把每个节点视为跟节点向下遍历统计路径和。当我们遍历到跟节点之后,节点为 null,此时返回 0,然后再计算 targetSum 与当前节点之差,如果差值等于 0 则统计结果加一,再继续递归遍历节点的左右子树,最后返回统计结果。
2.2 代码
class Solution {
public int pathSum(TreeNode root, int targetSum) {
if (root == null) {
return 0;
}
int ans = process(root, targetSum);
ans += pathSum(root.left, targetSum);
ans += pathSum(root.right, targetSum);
return ans;
}
public int process(TreeNode root, long target) {
if (root == null) {
return 0;
}
int ans = 0;
long val = target - root.val;
if (val == 0) {
ans++;
}
ans+=process(root.left,val);
ans+=process(root.right,val);
return ans;
}
}
2.3 测试结果
通过测试
3.总结
- 把每个节点视为跟节点遍历数组,并计算路径和