437. 路径总和 III

提示：文章写完后，目录可以自动生成，如何生成可参考右边的帮助文档

---

题目描述

给定一个二叉树，它的每个结点都存放着一个整数值。

找出路径和等于给定数值的路径总数。

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

二叉树不超过1000个节点，且节点数值范围是 [-1000000,1000000] 的整数。

示例：

root = [10,5,-3,3,2,null,11,3,-2,null,1], sum = 8

返回 3。和等于 8 的路径有:

1. 5 -> 3
2. 5 -> 2 -> 1
3. -3 -> 11

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/path-sum-iii
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

解题过程

解题思路

利用前缀和，将从根节点出发到当前节点的前缀和记为prefixSum。有这样的关系：
当前节点的前缀和与历史上某个节点的前缀和之差==targetSum，说明这两个节点之间的路径满足条件。因此我们将前缀和作为key，当前节点之前的前缀和的数量为value，用一个HashMap装起来。
需要注意的有两点：
1、前缀和为0的value值要默认为1；
2、map要回溯，不然会影响其他层。

/**
 * 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) {
        HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
        map.put(0, 1);
        return recur(root, map, 0, targetSum);

    }

    public int recur(TreeNode root, HashMap<Integer, Integer> map, int prefixSum, int targetSum){
        if(root == null){
            return 0;
        }
        int res = 0;
        prefixSum += root.val;

        res += map.getOrDefault(prefixSum - targetSum, 0);
        map.put(prefixSum, map.getOrDefault(prefixSum, 0) + 1);//把当前符合的路径加1
        //递归进入下一层
        res += recur(root.left, map, prefixSum, targetSum);
        res += recur(root.right, map, prefixSum, targetSum);
        //要回溯，不要让本层影响其他层
        map.put(prefixSum, map.getOrDefault(prefixSum, 0) - 1);
        return res;

    }
}

总结