【LeetCode刷题笔记Java】437.路径总和3

最新推荐文章于 2024-12-02 22:00:00 发布

C-Liueasymoney

最新推荐文章于 2024-12-02 22:00:00 发布

阅读量290

点赞数

分类专栏： LeetCode 树文章标签： java 算法 leetcode hadoop dfs

本文链接：https://blog.youkuaiyun.com/weixin_42050795/article/details/118491943

版权

LeetCode 同时被 2 个专栏收录

46 篇文章

订阅专栏

树

7 篇文章

订阅专栏

这篇博客主要探讨了使用Java解决LeetCode中的437题——路径总和3的方法，重点讲解了前缀和、递归以及回溯的策略。通过这些技术，博主详细解释了解题思路并提供了代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

public class MySolution {
    public int pathSum(TreeNode root, int targetSum){
        // key存放前缀和是几，value存放前缀和为key的有几个
        Map<Integer, Integer> prefixSumMap = new HashMap<>();
        // 加一个前缀和为0的，代表当前节点,也就是从根节点到根节点的前缀和
        prefixSumMap.put(0, 1);
        return process(root, prefixSumMap, targetSum, 0);
    }

    // cur代表从根节点走到当前节点的前缀和
    public int process(TreeNode root, Map<Integer, Integer> prefixSumMap, int targetSum, int cur){
        // case base
        if (root == null)
            return 0;
        // res代表当前节点上能够找到的满足targetSum的路径数目
        int res = 0;
        // 当前路径和先加上当前节点的值，表示走到了此节点
        cur += root.val;
        // 由于两节点的前缀和之差等于两节点的路径和，那么要找到路径和等于targetSum的路径
        // 就是要找当前节点的前缀和减去targetSum有没有存在map中，就表示之前有没有出现过路径和为cur-targetSum的节点
        res += prefixSumMap.getOrDefault(cur - targetSum, 0);
        // 把当前节点的路径和保存在map中
        prefixSumMap.put(cur, prefixSumMap.getOrDefault(cur, 0) + 1);

        // 对左右节点的结果进行收集
        res += process(root.left, prefixSumMap, targetSum, cur);
        res += process(root.right, prefixSumMap, targetSum, cur);

        // 回溯要对map中保存的当前节点前缀和进行消除
        prefixSumMap.put(cur, prefixSumMap.get(cur) - 1);

        return res;
    }
}