题目
题目链接:力扣437:路经总和Ⅲ
给定一个二叉树的根节点 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
提示:
- 二叉树的节点个数的范围是 [0,1000]
- -109 <=
Node.val
<= 109 - -1000 <=
targetSum
<= 1000
解题思路
建议先去写力扣112:路径总和 和 力扣113:路径总和Ⅱ 再看这题。循序渐进嘛。
👉双重递归:
如果写了上面两题就不难想到这题需要加一个函数的递归就可以解决。
首先,我们可以先写一个用来判断根节点到其他节点的路径和是否等于目标值 targetSum
的函数。这个函数应该不难写,直接在前序遍历的基础上加上一个判断:累加到目前节点的路径和是否等于目标值
- 等于时,计数变量加一
- 不等于时,继续递归
有了这个函数,然后让调用该函数的函数自己进行前序递归。这样可以让根节点不断变化到每个节点,这样就可以遍历到每种情况了。
但这种方法的缺点是,重复次数太多。每更换一次根节点就要重新计算一次根节点后面的节点路径和,这里的进阶版就是前缀和写法。
👉前缀和
前缀和最大的好处是避免了重复计算,就是说遍历了一次节点,就可以记录下该节点所有的情况。
这里当然少不了哈希表了,利用哈希表存储算是常见的了。注意题目给的节点值的范围,所以累加起来可能会爆 int
,最保险的是用 long long
。既然是遍历所有节点,当然可以使用上一种方法里的前序遍历了。在遍历的过程中,需要做的是:
- 累加上当前节点的值
- 判断哈希表里是否有
sum - targetSum
,或者说之前有没有存储过sum - targetSum
这个值。
这个判断条件是因为,如果之前存储过这个值,那么从根节点到目前的节点的路径和就是 sum
,拿 sum
减去存储过的值,就是目标值 targetSum
。从路径和为 sum - targetSum
的节点到当前节点的路径和就是目标值。
- 将累加的路径和存入哈希表中,递归函数调用完了需要恢复,即将其移除。
具体不详可以参考代码
代码(C++)
双重递归
class Solution {
public:
int f, sum;
void dfs(TreeNode *root, int targetSum) {
if (!root)
return ;
sum += root->val;
if (sum == targetSum)
f ++;
dfs(root->left, targetSum);
dfs(root->right, targetSum);
sum -= root->val;
}
int pathSum(TreeNode* root, int targetSum) {
if (!root)
return 0;
sum = 0;
dfs(root, targetSum);
pathSum(root->left, targetSum);
pathSum(root->right, targetSum);
return f;
}
};
前缀和
class Solution {
public:
unordered_map<long long, int> map;
int dfs(TreeNode *root, int targetSum, int sum) {
if (!root)
return 0;
sum += root->val;
int res = 0;
if (map.count(sum - targetSum))
res = map[sum - targetSum];
map[sum] ++;
res += dfs(root->left, targetSum, sum);
res += dfs(root->right, targetSum, sum);
map[sum] --;
return res;
}
int pathSum(TreeNode* root, int targetSum) {
if (!root)
return 0;
map[0] = 1;
return dfs(root, targetSum, 0);
}
};
总结
路径总和的前两题难度不大,所以就直接讲了这题。
这题我没有写出来,双重递归的思路很简单,没怎么遇到过这类题就没有想到双重递归。前缀和的优化其实也没有那么难,总体来说不能算简单吧。多练练这类题,下次遇到应该就会顺手一些。