问题:
给定二叉树的根节点 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
解答思路:
以下是使用 Java 实现找出所有从根节点到叶子节点路径总和等于给定目标和的路径的代码:
import java.util.ArrayList;
import java.util.List;
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int val) {
this.val = val;
}
}
public class PathSumII {
public static List<List<Integer>> pathSum(TreeNode root, int targetSum) {
List<List<Integer>> result = new ArrayList<>();
dfs(root, targetSum, new ArrayList<>(), result);
return result;
}
private static void dfs(TreeNode root, int targetSum, List<Integer> path, List<List<Integer>> result) {
if (root == null) {
return;
}
path.add(root.val);
if (root.left == null && root.right == null && targetSum == root.val) {
result.add(new ArrayList<>(path));
}
dfs(root.left, targetSum - root.val, path, result);
dfs(root.right, targetSum - root.val, path, result);
path.remove(path.size() - 1);
}
public static void main(String[] args) {
// 构建二叉树
TreeNode root = new TreeNode(5);
root.left = new TreeNode(4);
root.right = new TreeNode(8);
root.left.left = new TreeNode(11);
root.left.left.left = new TreeNode(7);
root.left.left.right = new TreeNode(2);
root.right.left = new TreeNode(13);
root.right.right = new TreeNode(4);
root.right.right.left = new TreeNode(5);
root.right.right.right = new TreeNode(1);
int targetSum = 22;
List<List<Integer>> paths = pathSum(root, targetSum);
for (List<Integer> path : paths) {
System.out.println(path);
}
}
}
上述代码中,我们首先定义了一个'TreeNode'类来表示二叉树的节点。然后,在'pathSum'方法中,我们使用深度优先搜索(DFS)的方式遍历二叉树。在每次递归中,我们将当前节点的值添加到路径中,并检查是否满足目标和。如果满足,我们将当前路径添加到结果列表中。然后,我们继续递归遍历左子树和右子树,并在回溯时将当前节点从路径中移除。
在'main'方法中,我们构建了一个示例二叉树,并指定了目标和。然后,我们调用'pathSum'方法获取所有满足条件的路径,并打印出来。
(文章为作者在学习java过程中的一些个人体会总结和借鉴,如有不当、错误的地方,请各位大佬批评指正,定当努力改正,如有侵权请联系作者删帖。)