给你二叉树的根节点 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
思路
思路和LeetCode——112 路径总和(JAVA)大致相似,dfs搜索的终止条件在于当前结点是叶子结点,然后此时判断路径上的所有点权之和是否等于给定的targetSum
,如果等于,将这条路加入最终的path
(双层List
嵌套)。
【注】一定要理解Java中一切皆对象的概念,也就是说,除了基本数据类型是值传递,其他都是地址传递。所以,在将当前路加入path
中时,一定要new
一个新对象,然后用for
循环逐个将tempPath
里的值给这个new
出来的新对象才行(如果直接用等于号赋值,那么两个对象指向的是同一个内存地址)。
代码
public class Solution {
public static class TreeNode {//为了方便本地调试调用内部类,我写了静态的
//实际提交时可以不写TreeNode类
int val;
TreeNode left;
TreeNode right;
TreeNode() {}
TreeNode(int val) { this.val = val; }
TreeNode(int val, TreeNode left, TreeNode right) {
this.val = val;
this.left = left;
this.right = right;
}
}
ArrayList<Integer> tempPath = new ArrayList<Integer>();
List<List<Integer>> path = new ArrayList<List<Integer>>();
public void dfs(TreeNode root, int targetSum) {
tempPath.add(root.val);
if(root.left==null&&root.right==null) {//如果当前是叶子结点
int sum = 0;
for(int i=0;i<tempPath.size();i++) {
sum += tempPath.get(i);
}
if(sum==targetSum) {
ArrayList<Integer> tmp = new ArrayList<Integer>();//要new一个新对象来放
//否则path里面的几次add都是添加的tempPath这同一个对象
//因此当tempPath本身内容改变时,path里放进去的tempPath也会改变(引用传递,非值传递)
for(int i=0;i<tempPath.size();i++) {
tmp.add(tempPath.get(i));
}
path.add(tmp);
}
return;
}
if(root.left!=null) {
dfs(root.left, targetSum);
tempPath.remove(tempPath.size()-1);
}
if(root.right!=null) {
dfs(root.right, targetSum);
tempPath.remove(tempPath.size()-1);
}
}
public List<List<Integer>> pathSum(TreeNode root, int targetSum) {
if(root!=null) {
dfs(root, targetSum);
}
return path;
}
}