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

发布时间:2023年12月28日

花无缺

👨?💻博客主页:@花无缺
欢迎 点赞👍 收藏? 留言📝 加关注?!
本文由 花无缺 原创

收录于专栏 【力扣题解】



【力扣题解】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万字详解】
你们的点赞👍 收藏? 留言📝 关注?
是我持续创作,输出优质内容的最大动力!
谢谢!

文章来源:https://blog.csdn.net/huawuque404/article/details/135274774
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。