二叉树-路径总和III

Vacant Seat2025-03-19 11:11

java 复制代码

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

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

输入：二叉树根结点，目标数
输出：整型变量
思路：使用前缀和，然后哈希表维护二叉树

java 复制代码

/**
 * 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<>();
        //将（0,1）存入
        map.put(0L, 1);
        //递归遍历
        return dfs(root, map, 0L, targetSum);
    }
    public int dfs(TreeNode root, Map<Long, Integer> map, Long sum, int targetSum){
        //避免空指针异常，递归结束
        if(root == null){
            return 0;
        }
        int count = 0;
        sum += root.val;
        //记录当前sum - targetSum出现过几次
        count = map.getOrDefault(sum - targetSum, 0);
        //然后将当前前缀和加入map
        map.put(sum, map.getOrDefault(sum, 0) + 1);
        //递归左右子树
        count += dfs(root.left, map, sum, targetSum);
        count += dfs(root.right, map, sum, targetSum);
        //然后将退回上一层
        map.put(sum, map.getOrDefault(sum, 0) - 1);
        return count;
    }
}