【力扣题解】P113-路径总和 II-Java题解

‍博客主页：@花无缺
欢迎 点赞 收藏⭐ 留言 加关注✅!
本文由 花无缺 原创

收录于专栏 【力扣题解】

---

---

【力扣题解】P113-路径总和 II-Java题解

P113.路径总和 II

题目描述

给你二叉树的根节点 root 和一个整数目标和 targetSum ，找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。

叶子节点 是指没有子节点的节点。

示例 1：

输入：root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22
输出：[[5,4,11,2],[5,8,4,5]]

示例 2：

输入：root = [1,2,3], targetSum = 5
输出：[]

示例 3：

输入：root = [1,2], targetSum = 0
输出：[]

提示：

• 树中节点总数在范围 [0, 5000] 内
• -1000 <= Node.val <= 1000
• -1000 <= targetSum <= 1000

题解

深度优先搜索递归：

class Solution {
    // path 存储每一条遍历的路径
    List<Integer> path = new ArrayList<>();
    // res 存储满足条件的路径
    List<List<Integer>> res = new ArrayList<>();
    public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
        // 空树, 返回空列表
        if (root == null) return res;
        // 将根节点值加入 path 路径中
        path.add(root.val);
        dfs(root, targetSum - root.val);
        return res;
    }
    public void dfs(TreeNode node, int count) {
        // 如果当前节点(根节点)是叶子节点并且路径和等于 targetSum
        // 那么将路径加入结果列表 res
        if (node.left == null && node.right == null && count == 0) {
            res.add(new ArrayList<>(path));
            return;
        }
        // 如果路径和不等于 targetSum, 递归终止
        if (node.left == null && node.right == null) {
            return;
        }
        // 递归左子节点
        if (node.left != null) {
            // 将当前节点值加入路径 path
            path.add(node.left.val);
            // count 值递减
            count = count - node.left.val;
            // 递归
            dfs(node.left, count);
            // 左子节点也没有找到符合的路径, 回溯
            // 移除左子节点值
            path.remove(path.size() - 1);
            // count 恢复原来的值
            count = count + node.left.val;
        }
        // 递归右子节点
        if (node.right != null) {
            // 将当前节点值加入路径 path
            path.add(node.right.val);
            // count 值递减
            count = count - node.right.val;
            // 递归
            dfs(node.right, count);
            // 右子节点也没有找到符合的路径, 回溯
            // 移除右子节点值
            path.remove(path.size() - 1);
            // count 恢复原来的值
            count = count + node.right.val;
        }
        // 整个树搜索完成, 递归终止
        return;
    }
}

时间复杂度：O(n)，需要遍历树的所有节点，树的节点数为 n。

总结

这个题目和P112-路径总和整体思路差不多，但是这个题多了一个我们需要维护的变量，在P112-路径总和中，我们只需要维护计数器 count 变量，并不断递归或者回溯，而本题我们不仅需要维护 count 变量，还要维护 path，表示当前路径的所有节点值列表，在遍历左右子树时，除了要回溯 count 变量，还要回溯 path 变量。

作者：花无缺(huawuque404.com)

---

欢迎关注我的博客：花无缺-每一个不曾起舞的日子都是对生命的辜负~
一起进步-刷题专栏：【力扣题解】
往期精彩好文：
【CSS选择器全解指南】
【HTML万字详解】
你们的点赞 收藏⭐ 留言 关注✅
是我持续创作，输出优质内容的最大动力！
谢谢！