lc437. 路径总和 III

最新推荐文章于 2024-06-20 16:42:54 发布

林中伊人

最新推荐文章于 2024-06-20 16:42:54 发布

阅读量110

点赞数

分类专栏：算法 java leetcode 文章标签：深度优先算法 java

本文链接：https://blog.csdn.net/weixin_45686265/article/details/134092347

版权

java 同时被 3 个专栏收录

19 篇文章 0 订阅

订阅专栏

算法

16 篇文章 0 订阅

订阅专栏

leetcode

15 篇文章 0 订阅

订阅专栏

两种解法

解法一：直接两个递归，但是重复的计算过多

解法二：前缀和求解！

package com.codeking.lc;

import com.codeking.Node.TreeNode;

import java.util.HashMap;
import java.util.Map;

/**
 * @author xiongjl
 * @since 2023/10/28  14:57
 */
public class lc437 {

    // 前缀和解法
    public int pathSum(TreeNode root, int targetSum) {
        // 考虑根节点
        map.put(0L, 1);
        DFS(root, targetSum, 0);
        return cnt;
    }

    Map<Long, Integer> map = new HashMap<>();
    int cnt = 0;

    void DFS(TreeNode node, long targetSum, long preSum) {
        if (node == null) {
            return;
        }
        preSum += node.val;
        cnt += map.getOrDefault(preSum-targetSum, 0);
        // 前缀和
        map.put(preSum, map.getOrDefault(preSum, 0) + 1);
        DFS(node.left, targetSum, preSum);
        DFS(node.right, targetSum, preSum);
        // 回溯
        map.put(preSum, map.get(preSum)- 1);
    }

   /* public int pathSum(TreeNode root, int targetSum) {
        if (root == null) {
            return 0;
        }
        int ret = DFS(root, targetSum);
        ret += pathSum(root.left, targetSum);
        ret += pathSum(root.right, targetSum);
        return ret;
    }*/

    /*int DFS(TreeNode node, long targetSum) {
        int cnt = 0;
        if (node == null) {
            return 0;
        }
        int val = node.val;
        if (val == targetSum) {
            cnt++;
        }
        cnt += DFS(node.left, targetSum - val);
        cnt += DFS(node.right, targetSum - val);
        return cnt;
    }*/
}