详细注释的代码在下方
var pathSum = function(root, targetSum) {
let perfix = new Map();
perfix.set(0, 1);
function dfs(curr,root){
if(!root) return 0;
let ret = 0;
curr += root.val;
ret += perfix.get(curr - targetSum) || 0;
perfix.set(curr, (perfix.get(curr) || 0) + 1);
ret += dfs(curr, root.left);
ret += dfs(curr, root.right);
perfix.set(curr, perfix.get(curr) - 1);
return ret;
}
return dfs(0, root);
}
详细注释版
var pathSum = function(root, targetSum) {
const perfix = new Map();
perfix.set(0,1);
/*初始化:在开始遍历树之前,我们需要一个初始值来表示路径的前缀和为 0。这是因为在从根节点到某个节点的路径上,如果从根节点出发,没有任何节点,则路径的前缀和为 0。
第一个节点:考虑第一个节点的情况,如果第一个节点的值恰好等于目标和 targetSum,那么这个节点本身就可以构成一条路径。而这个节点到根节点的路径的前缀和为 0,因此我们需要将前缀和为 0 的出现次数初始化为 1,以确保能够正确地计算出路径的数量。所以,在开始遍历树之前,我们将前缀和为 0 的出现次数设为 1,以确保算法能够正确地处理路径的情况。*/
function dfs(root,curr){
if(!root) return 0;
let ret = 0;//用于存储从当前节点开始的满足路径总和等于目标和的路径数
curr += root.val;//累加前缀和
ret += perfix.get(curr - targetSum) || 0;//判断有没有满足的前缀和
perfix.set(curr,(perfix.get(curr - targetSum) || 0) + 1);//向map添加当前值的前缀和
ret += dfs(root.left,curr);
ret += dfs(root.right,curr);
perfix.set(curr,perfix.get(curr) - 1);
/*将 perfix.set(curr,(perfix.get(curr) || 0) - 1); 放到末尾的目的是在当前节点的递归处理结束后,及时将当前前缀和 curr 的出现次数减 1。这样可以保证在回溯到父节点时,前缀和 Map 中记录的信息是正确的,不会影响到父节点及其后续的处理。
具体来说,这行代码的执行时机在于当前节点的递归处理结束后,即当前节点的左右子树都已经递归处理完毕,返回到了当前节点的父节点。在这个时候,我们需要将当前节点的前缀和从前缀和 Map 中去除,以确保下一次遇到同样的前缀和时,能够得到正确的计数
举例,root =[1,-2,-3](根,左子节点,右子节点),targetSum =-1
左子树的遍历中,在左子节点处,curr = 1 + (-2) = -1,此时map中存入键值,这个是左子树的前缀和,
回到右子树中,在右子节点-3处,curr = 1+(-3)=-2,ret += perfix.get(curr - targetSum),此时会加上左子树中加入的-1,最终结果就是左右子树个返回1个前缀和,结果就不对了,所以要对当前遍历后的结果进行回溯*/
return ret;
}
};