leetcode437. 路径总和 III !

最新推荐文章于 2024-08-24 13:50:23 发布

Rookie2Master

最新推荐文章于 2024-08-24 13:50:23 发布

阅读量196

点赞数

分类专栏： leetcode 文章标签：二叉树 leetcode 回溯

本文链接：https://blog.csdn.net/qq_43778308/article/details/106298023

版权

leetcode 专栏收录该内容

227 篇文章 1 订阅

订阅专栏

传送门

给定一个二叉树，它的每个结点都存放着一个整数值。找出路径和等于给定数值的路径总数。
路径不需要从根节点开始，也不需要在叶子节点结束，但是路径方向必须是向下的

Tree：10         sum = 8                
     /  \
    5   -3
   / \    \
  3   2   11
 / \   \
3  -2   1

返回 3。和等于 8 的路径有:  5 -> 3     5 -> 2 -> 1     -3 -> 11

方法1. 双递归

最暴力方法：以每个节点为根节点，全都算一遍路径和为sum的有几条，然后加起来。

	public int pathSum(TreeNode root, int sum) {
        if (root == null) return 0;
        return helper(root, sum) +
               pathSum(root.left, sum) +
               pathSum(root.right, sum);
    }
    public int helper(TreeNode root, int sum) {
        if (root == null) return 0;
        sum -= root.val;
        int ans = sum == 0 ? 1 : 0;
        return ans + helper(root.left, sum) + helper(root.right, sum);
	}

方法2. DFS回溯+记忆化搜索

前缀和:

如果两个数的前缀总和是相同的，那么这些节点之间的元素总和为零。进一步扩展相同的想法，如果前缀总和currSum，在节点A和节点B处相差target，则位于节点A和节点B之间的元素之和是target。

因为本题中的路径是一棵树，从根往任一节点的路径上(不走回头路)，有且仅有一条路径，因为不存在环。

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

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

注意：put进去的是前缀和，是没有减去sum的，查找的时候当前的pathSum需要减去sum 这样才能得到另一hash表里的前缀和。（画个图）

为什么hash表先 put(0,1)?

hash的key代表路径和，那么0->1 是天然存在的，就是说：还没有开始遍历，前缀和就是0
比如：树只有节点10, sum也是10，遍历完root之后pathsum-sum=0 这样就要用到这个前缀和0了

	int target = 0; // 输入的sum
    Map<Integer, Integer> hashPreSum = new HashMap<>();
    public int pathSum(TreeNode root, int sum) {
        if (root == null) return 0;
        target = sum;
        hashPreSum.put(0, 1);
        return helper(root, 0);
    }
    public int helper(TreeNode root, int pathSum) {
        if (root == null) return 0;
        pathSum += root.val;
        int preSum = pathSum - target;//当前路径和 - target = map里存的前缀和
        int ans = hashPreSum.getOrDefault(preSum, 0); //不能放在put后面: 输入root=1，sum=0
        hashPreSum.put(pathSum, hashPreSum.getOrDefault(pathSum, 0) + 1);
        ans +=  helper(root.left, pathSum) + helper(root.right, pathSum);
        hashPreSum.put(pathSum, hashPreSum.get(pathSum) - 1); // 要回溯
        return ans;
	}