👨?💻博客主页:@花无缺
欢迎 点赞👍 收藏? 留言📝 加关注?!
本文由 花无缺 原创收录于专栏 【力扣题解】
给你二叉树的根节点 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 变量。
🌸欢迎
关注
我的博客:花无缺-每一个不曾起舞的日子都是对生命的辜负~
🍻一起进步-刷题专栏:【力扣题解】
🥇往期精彩好文:
📢【CSS选择器全解指南】
📢【HTML万字详解】
你们的点赞👍 收藏? 留言📝 关注?
是我持续创作,输出优质内容
的最大动力!
谢谢!