向下的路径节点之和

最新推荐文章于 2022-09-18 17:24:31 发布

炸娃（java）

最新推荐文章于 2022-09-18 17:24:31 发布

阅读量170

点赞数

分类专栏： java 数据结构文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/zyf7764029/article/details/125868368

版权

数据结构同时被 2 个专栏收录

12 篇文章 0 订阅

订阅专栏

java

9 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何解决一个二叉树问题，即找到所有节点值之和等于给定目标值的路径。文章提出了两种解法，一种是从根节点向下搜索，另一种是从当前节点向上搜索，利用哈希映射统计路径。这两种方法都涉及到递归和状态跟踪，对于理解二叉树遍历和路径查找有很好的启示作用。

摘要由CSDN通过智能技术生成

给定一个二叉树的根节点 root ，和一个整数 targetSum ，求该二叉树里节点值之和等于 targetSum 的路径的数目。路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的（只能从父节点到子节点）。题目链接：
这题有两种解法，第一种是从根节点向叶节点搜索，此时有两种情况，考虑当前节点和不考虑当前节点。考虑当前节点要一直搜索到叶子节点，因为节点的值有的为负数。例如targeSum = 15，
在这里插入图片描述

class Solution {
    public int pathSum(TreeNode root, int targetSum) {
        int ret = 0;
        if(root == null)
            return ret;
        ret += rootSum(root, targetSum);
        ret += pathSum(root.left, targetSum);
        ret += pathSum(root.right, targetSum);
        return ret;
    }

    public int rootSum(TreeNode root, int targetSum){
        int ret = 0;
        if(root == null)
            return ret;
        if(root.val == targetSum)
            ret++;
        ret += rootSum(root.left, targetSum - root.val);
        ret += rootSum(root.right, targetSum - root.val);
        return ret;
    }
}

还有一种是从当前节点向上搜索，统计从当前节点线向上有多少条路径的和等于targetSum。对如targetSum=15，如下图，对于起始节点，我们要找有多少路径等于targetSum，也就等价于找有多少个preSum = Sum - targetSum，也就是统计Sum-targetSum出现的次数。

在这里插入图片描述

class Solution {
    public int pathSum(TreeNode root, int targetSum) {
        Map<Integer, Integer> map = new HashMap<Integer, Integer>();
        map.put(0, 1);
        return path(root, map, 0, targetSum);
    }

    public int path(TreeNode root, Map<Integer, Integer> map, int cur, int targetSum){
        if(root == null)
            return 0;
        cur += root.val;
        int ret = map.getOrDefault(cur - targetSum, 0);
        map.put(cur, map.getOrDefault(cur, 0) + 1);
        ret += path(root.left, map, cur, targetSum);
        ret += path(root.right, map, cur, targetSum);
        map.put(cur, map.get(cur) - 1);
        return ret;
    }
}