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;
    }
}
相关推荐
CoderCodingNo30 分钟前
【GESP】C++五级练习题 luogu-P1865 A % B Problem
开发语言·c++·算法
大闲在人40 分钟前
7. 供应链与制造过程术语:“周期时间”
算法·供应链管理·智能制造·工业工程
小熳芋43 分钟前
443. 压缩字符串-python-双指针
算法
Charlie_lll1 小时前
力扣解题-移动零
后端·算法·leetcode
chaser&upper1 小时前
矩阵革命:在 AtomGit 解码 CANN ops-nn 如何构建 AIGC 的“线性基石”
程序人生·算法
weixin_499771551 小时前
C++中的组合模式
开发语言·c++·算法
iAkuya2 小时前
(leetcode)力扣100 62N皇后问题 (普通回溯(使用set存储),位运算回溯)
算法·leetcode·职场和发展
近津薪荼2 小时前
dfs专题5——(二叉搜索树中第 K 小的元素)
c++·学习·算法·深度优先
xiaoye-duck2 小时前
吃透 C++ STL list:从基础使用到特性对比,解锁链表容器高效用法
c++·算法·stl
松☆2 小时前
CANN与大模型推理:在边缘端高效运行7B参数语言模型的实践指南
人工智能·算法·语言模型