hot100 437.路径总和Ⅲ

思路:前缀和解法,利用前缀和求节点值之和等于targetSum的路径的数目(满足路径不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的,只能从父节点到子节点)。

一、前缀和定义:

1.定义:一个节点的前缀和就是该节点到根之间的路径和。

2.举例:如下图所示。

(1)节点4的前缀和为:1 + 2 + 4 = 7。

(2)节点8的前缀和为:1 + 2 + 4 + 8 = 15。

(3)节点9的前缀和为:1 + 2 + 5 + 9 = 17。

二、本题中前缀和的作用:两节点间的路径和即为两节点的前缀和之差。

1.举例:

(1)如下图所示:假如题目给定数值为5,节点1的前缀和为1,节点3的前缀和为1 + 2 + 3 = 6。

(2)prefix(3) - prefix(1) == 5。

(3)所以节点1到节点3之间,有一条符合要求的路径2->3。

2.利用前缀和简化问题:我们只需要遍历整棵树一次,记录每个节点的前缀和,并查询该节点的祖先节点中符合条件的个数,将这个数量加到最终结果上。

三、HashMap存储数据:key存储前缀和,value是该前缀和的节点数量,记录数量是因为有出现重复路径的可能。

举例:如下图所示,改图中前缀和为1的节点有两个,分别是0和1,所以路径和为2的路径数就有两条,分别是0->2和2。

.

四、恢复状态的意义:

1.题目要求:路径方向必须是向下的,即只能从父节点到子节点。当计算两个节点的前缀和的差值时,其中一个节点必须是另一个节点的祖先节点。

2.换句话说,当我们把一个节点的前缀和信息更新到map里时,它应当只对其子节点们生效。

举例:下图中有两个值为2的节点A,B。

(1)当遍历到最右方的节点6时,对于它来说,此时的前缀和为2的节点只该有B,因为A并不是节点6的祖先节点。

(2)如果不做状态恢复的话,当遍历右子树时,左子树中A的信息仍会保留在map中,那么此时节点6就会认为A,B都是可追溯到的节点,从而产生错误。

3.状态恢复代码的作用:在遍历完一个节点的所有子节点后,将其从map中除去。

五、核心思想:寻找从根节点到当前节点的路径上,是否存在某个前缀和curSum - target。如果存在,则说明从那个前缀和对应的节点到当前节点的路径和正好等于target。

即:curSum - (curSum - target) == target。

附代码:

java 复制代码
class Solution {
    Map<Long,Integer> prefixMap;
    int target;
    public int pathSum(TreeNode root, int targetSum) {
        prefixMap = new HashMap<>();
        target = targetSum;

        // key表示前缀和,value表示前缀和为这个值的路径的数量。
        // 初始化,表示空路径的前缀和为0,出现了1次,为了处理从根节点开始的路径
        prefixMap.put(0L,1); 
        return recur(root,0L);
    }

    // recur函数,用于遍历二叉树,对于每个节点,记录前缀和的同时,返回当前找到和为target的路径的数量
    // curSum表示node节点之前的前缀和
    private int recur(TreeNode node,Long curSum){
        if(node == null){
            return 0;
        }
        int res = 0; // 记录路径数量
        curSum += node.val; // 根节点到当前节点的路径和

        // 如果curSum - target在prefixMap中存在,则说明存在某个节点到当前节点的路径和等于target
        // curSum - (curSum - target) = target
        // 比如:target = 8,如果之前存在前缀和 = 5,且当前前缀和 = 13,那么13 - 5 = 8,说明从那个节点到当前节点的路径和为8
        res += prefixMap.getOrDefault(curSum - target,0);
        // 把当前节点的前缀和也加入到哈希表中,value加1
        prefixMap.put(curSum,prefixMap.getOrDefault(curSum,0) + 1);

        // 递归处理子树
        int left = recur(node.left,curSum);
        int right = recur(node.right,curSum);

        res = res + left + right; // 所有找到的路径数量总和即为总的路径数量

        // 恢复状态
        prefixMap.put(curSum,prefixMap.get(curSum) - 1); // 当前节点处理完后,要返回到父节点,当前节点的前缀和不应该影响其他分支的统计。因此在遍历完一个节点的所有子节点后,将其从map中除去
        return res;
    }
}
相关推荐
6Hzlia5 小时前
【Hot 100 刷题计划】 LeetCode 739. 每日温度 | C++ 逆序单调栈
c++·算法·leetcode
良木生香5 小时前
【C++初阶】:STL——String从入门到应用完全指南(1)
c语言·开发语言·数据结构·c++·算法
XWalnut6 小时前
LeetCode刷题 day16
数据结构·算法·leetcode·链表·动态规划
foundbug9997 小时前
基于混合整数规划的电池容量优化 - MATLAB实现
数据结构·算法·matlab
memcpy09 小时前
LeetCode 2452. 距离字典两次编辑以内的单词【暴力;字典树】中等
算法·leetcode·职场和发展
王老师青少年编程9 小时前
csp信奥赛C++高频考点专项训练之贪心算法 --【排序贪心】:魔法
c++·算法·贪心·csp·信奥赛·排序贪心·魔法
wearegogog1239 小时前
基于和差波束法的单脉冲测角MATLAB实现
人工智能·算法·matlab
AI科技星9 小时前
灵魂商数(SQ) · 全域数学统一定义【乖乖数学】
算法·机器学习·数学建模·数据挖掘·量子计算
晓觉儿9 小时前
【GPLT】2026年第十一届团队程序设计天梯赛赛后题解(已写2h,存档中)
数据结构·c++·算法·深度优先·图论