解题思路:首先判断根结点是否为空,不为空的话判断当前结点是否是叶子结点;如果当前存入集合的值小于target,且有左右子树,则递归;如果当前存入集合的值等于target,且为叶子结点,则打印路径;如果当前存入集合的值大于target,则不用递归了。(题目中,默认所有结点的值都为整数)
代码如下:
import java.util.ArrayList;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
ArrayList<ArrayList<Integer>> list = new ArrayList<>();
ArrayList<Integer> arr = new ArrayList<>();
int num =0;
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root==null)
return list;
boolean leaf = root.left==null && root.right==null;
num+=root.val;
arr.add(root.val);
if(num==target && leaf){
ArrayList<Integer> path = new ArrayList<>();
for(int i=0;i<arr.size();i++){
path.add(arr.get(i));
}
list.add(path);
}
if(num<target && root.left!=null){
FindPath(root.left, target);
}
if(num<target && root.right!=null){
FindPath(root.right, target);
}
//当前节点值的和大于target,那就没必要递归,同时把最后添加的值给移除
num-=root.val;
arr.remove(arr.size()-1);
return list;
}
}