一刷194-力扣热题-437路径总和III(m)(同:剑指 Offer II 050. 向下的路径节点之和)

题目:
给定一个二叉树的根节点 root ,和一个整数 targetSum ,
求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始,也不需要在叶子节点结束,
但是路径方向必须是向下的(只能从父节点到子节点)。

在这里插入图片描述

输入:root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8
输出:3
解释:和等于 8 的路径有 3 条,如图所示。

输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出:3
 
提示:
二叉树的节点个数的范围是 [0,1000]
-109 <= Node.val <= 109 
-1000 <= targetSum <= 1000 
----------------------------------
思考:
这道题用到了一个概念,叫前缀和。就是到达当前元素的路径上,之前所有元素的和。
前缀和怎么应用呢?

在同一个路径之下(可以理解成二叉树从root节点出发,到叶子节点的某一条路径),
如果两个数的前缀总和是相同的,那么这些节点之间的元素总和为零。
进一步扩展相同的想法,
如果前缀总和currSum,在节点A和节点B处相差target,则位于节点A和节点B之间的元素之和是target。

因为本题中的路径是一棵树,从根往任一节点的路径上(不走回头路),有且仅有一条路径,因为不存在环。
(如果存在环,前缀和就不能用了,需要改造算法)

抵达当前节点(B节点)后,将前缀和累加,然后查找在前缀和上,
有没有前缀和currSum-target的节点(A节点),
存在即表示从AB有一条路径之和满足条件的情况。
结果加上满足前缀和currSum-target的节点的数量。然后递归进入左右子树。

左右子树遍历完成之后,回到当前层,需要把当前节点添加的前缀和去除。
避免回溯之后影响上一层。因为思想是前缀和,不属于前缀的,我们就要去掉它。

代码:
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public int pathSum(TreeNode root, int targetSum) {
        Map<Long, Integer> map = new HashMap<>();//key:前缀和 value:大小为key的前缀和出现的次数
        map.put(0L, 1);//前缀和为0的一条路径
        return backTracking(root, map, targetSum, 0); //前缀和的递归回溯思路
    }
    public int backTracking(TreeNode node, Map<Long, Integer> map, int targetSum, long curSum) {
        if (node == null) return 0;//边界条件 
        int res = 0;//结果集
        curSum += node.val;//当前路径和
        res += map.getOrDefault(curSum - targetSum, 0);
        map.put(curSum, map.getOrDefault(curSum, 0) + 1);//更新路径上的当前节点的前缀和个数 //!
        res += backTracking(node.left, map, targetSum, curSum);//进入下一层
        res += backTracking(node.right, map, targetSum, curSum);
        map.put(curSum, map.get(curSum) - 1);//回溯 //!
        return res;
    }
}

LC

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值