题目所属分类
路径系列的第三题 路径系列的题解如下
第一道题
第二道题
但是该题中对合法路径的要求是路径不需要从根节点开始,也不需要在叶子节点结束。所以我们可以枚举每个结点作为起点,同时在判断路径总和是否等于目标值时并不需要要求该点为叶子节点。
原题链接
给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。
路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。
代码案例:
输入:root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出:3
解释:和等于 8 的路径有 3 条,如图所示。
题解
y总的解法
哈希表中存放的是根节点到当前节点 前缀和为Si的个数
其中:
merge() 方法会先判断指定的 key 是否存在,如果不存在,则添加键值对到 hashMap 中。
merge() 方法的语法为:
hashmap.merge(key, value, remappingFunction)
注:hashmap 是 HashMap 类的一个对象。
参数说明:
key - 键
value - 值
remappingFunction - 重新映射函数,用于重新计算值
返回值
如果 key 对应的 value 不存在,则返回该 value 值,如果存在,则返回通过 remappingFunction 重新计算后的值。
注意案例中的long 换数据类型了 前缀和的适用场景:求任意不含环的结构,任意节点开头的路径和等于某个定值问题
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
//前缀和,前缀和出现的次数
Map<Long, Integer> map = new HashMap();
int res = 0;
public int pathSum(TreeNode root, int sum) {
map.put(0L, 1);
dfs(root, sum, 0L);
return res;
}
public void dfs(TreeNode root, int sum, long cur) {
if (root == null) return;
cur += root.val;
//如果此前有和为cur -sum,而当前的和又为cur ,两者的差就肯定为sum了
res += map.getOrDefault(cur - sum, 0);
map.put(cur, map.getOrDefault(cur, 0) + 1);
//下一层
dfs(root.left, sum, cur);
dfs(root.right, sum, cur);
//前缀和对于当前路径来说是唯一的,当前记录的前缀和,在回溯结束,回到本层时去除,保证其不影响其他分支的结果
//回到本层,恢复状态,去除当前节点的前缀和数量
map.put(cur, map.getOrDefault(cur, 0) - 1);
}
}
class Solution {
int res = 0;
// 存储每个前缀和出现的次数
Map<Long, Integer> cnt = new HashMap<>();
public int pathSum(TreeNode root, int targetSum) {
// 存一个哨兵
cnt.put(0l, 1);
dfs(root, targetSum, 0l);
return res;
}
public void dfs(TreeNode root, int targetSum, long preSum) {
if (root == null) return;
preSum += root.val;
// 寻找前面区间和为 si - sum 的区间的个数
res += cnt.getOrDefault(preSum - targetSum, 0);
cnt.merge(preSum, 1, Integer::sum);
dfs(root.left, targetSum, preSum);
dfs(root.right, targetSum, preSum);
// 回溯(只有一些全局变量才需要回溯)
cnt.merge(preSum, -1, Integer::sum);
}
}
单独测试这条案例能过 然后一提交就是出错
[1000000000,1000000000,null,294967296,null,1000000000,null,1000000000,null,1000000000]
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode() {}
* TreeNode(int val) { this.val = val; }
* TreeNode(int val, TreeNode left, TreeNode right) {
* this.val = val;
* this.left = left;
* this.right = right;
* }
* }
*/
class Solution {
public int pathSum(TreeNode root, int targetSum) {
if (null == root) return 0;
// 分别以root、root.left、root.right为根
return (int)dfs(root, targetSum) + pathSum(root.left, targetSum) + pathSum(root.right, targetSum);
}
public long dfs(TreeNode t, int s) {
long res = 0;
if (null == t) return 0;
if (s == t.val) res += 1;
res += dfs(t.left, s - t.val) + dfs(t.right, s - t.val);
return res;
}
}