LeetCode 437. 路径总和 III Path Sum III

本文详细解析了 LeetCode 上的 437. 路径总和 III 问题,介绍了如何使用递归算法在二叉树中查找路径和等于给定数值的所有路径,并提供了具体的 Java 实现代码。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

7-5 稍复杂的递归逻辑 Path Sum III

题目: LeetCode 437. 路径总和 III

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

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

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

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

示例:

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

  10
 /  \
5   -3

/ \
3 2 11
/ \
3 -2 1

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

  1. 5 -> 3
  2. 5 -> 2 -> 1
  3. -3 -> 11
/// 437. Path Sum III
/// https://leetcode.com/problems/path-sum-iii/description/
/// 时间复杂度: O(n), n为树的节点个数
/// 空间复杂度: O(h), h为树的高度
class Solution {

    /// Definition for a binary tree node.
    public static class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;
        TreeNode(int x) { val = x; }
    }

    // 在以root为根节点的二叉树中,寻找和为sum的路径,返回这样的路径个数
    public int pathSum(TreeNode root, int sum) {

        if(root == null)
            return 0;

        return findPath(root, sum)
                + pathSum(root.left , sum)
                + pathSum(root.right , sum);
    }

    // 在以node为根节点的二叉树中,寻找包含node的路径,和为sum
    // 返回这样的路径个数
    private int findPath(TreeNode node, int num){

        if(node == null)
            return 0;

        int res = 0;
        if(node.val == num)
            res += 1;

        res += findPath(node.left , num - node.val);
        res += findPath(node.right , num - node.val);

        return res;
    }

    public static void main(String[] args) {

        // 手动创建Leetcode题页上的测试用例。
        // 当然, 有更好的更智能的创建二叉树的方式, 有兴趣的同学可以自行研究编写程序:)

        /*****************
         * 测试用例:
         *
         *       10
         *      /  \
         *     5   -3
         *    / \    \
         *   3   2   11
         *  / \   \
         * 3  -2   1
         *****************/
        TreeNode node1 = new TreeNode(3);
        TreeNode node2 = new TreeNode(-2);

        TreeNode node3 = new TreeNode(3);
        node3.left = node1;
        node3.right = node2;

        TreeNode node4 = new TreeNode(1);
        TreeNode node5 = new TreeNode(2);
        node5.right = node4;

        TreeNode node6 = new TreeNode(5);
        node6.left = node3;
        node6.right = node5;

        TreeNode node7 = new TreeNode(11);
        TreeNode node8 = new TreeNode(-3);
        node8.right = node7;

        TreeNode node9 = new TreeNode(10);
        node9.left = node6;
        node9.right = node8;

        System.out.println((new Solution()).pathSum(node9, 8));
    }
}
### LeetCode Hot 100 路径总和 III Java 解决方案 #### 方法一:暴力递归法 此方法通过遍历每一个节点并尝试找到从该节点出发的所有可能路径,判断这些路径的和是否等于目标值。 ```java class Solution { int pathnumber; public int pathSum(TreeNode root, long sum) { if (root == null) return 0; Sum(root, sum); pathSum(root.left, sum); pathSum(root.right, sum); return pathnumber; } public void Sum(TreeNode root, long sum) { if (root == null) return; sum -= root.val; if (sum == 0) { pathnumber++; } Sum(root.left, sum); Sum(root.right, sum); } } ``` 这种方法虽然简单直观,但在处理大规模数据时效率较低。对于某些极端情况下的输入,可能会导致性能问题[^1]。 #### 方法二:优化后的前缀和加哈希表 为了提高算法效率,可以采用前缀和的概念加上哈希表来记录已经访问过的节点及其累积值。这样可以在一次深度优先搜索过程中完成计算,而不需要重复遍历子树。 ```java import java.util.HashMap; public class Solution { private HashMap<Long, Integer> prefixSumCount = new HashMap<>(); public int pathSum(TreeNode root, int targetSum) { prefixSumCount.put(0L, 1); return findPath(root, 0L, targetSum); } private int findPath(TreeNode node, long currentSum, int targetSum) { if (node == null) return 0; // 更新当前累计和 currentSum += node.val; // 计算满足条件的数量 int numPathsToCurrentNode = prefixSumCount.getOrDefault(currentSum - targetSum, 0); // 将当前累计和加入map中 prefixSumCount.put(currentSum, prefixSumCount.getOrDefault(currentSum, 0) + 1); // 继续向下探索左右子树 int leftResult = findPath(node.left, currentSum, targetSum); int rightResult = findPath(node.right, currentSum, targetSum); // 移除当前结点的影响以便回溯到父级调用者处继续其他分支的查找工作 prefixSumCount.put(currentSum, prefixSumCount.get(currentSum) - 1); return numPathsToCurrentNode + leftResult + rightResult; } } ``` 这种改进的方法不仅提高了时间复杂度至 O(n),而且空间上也更加高效,适用于更广泛的情况[^2]。 #### 数据约束说明 题目规定了二叉树中的节点数量范围以及各节点取值区间: - 二叉树的节点个数的范围是 [0,1000] - `-10^9 <= Node.val <= 10^9` - `-1000 <= targetSum <= 1000` 因此,在实现解决方案时需要注意数值类型的选取以防止溢出等问题的发生[^3]。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值