这道题目让我最难想到的是怎么保存符合条件的路径
1、满足条件的路径:从根节点到叶子节点的和==target的值
那么就用target-node.val如果减掉叶子节点的val后==0,就说明从root节点到这个leaf的路径满足要求
2、如何保存路径:
声明一个ArrayList前序遍历到某个节点的时候就add进去,然后判断一下是否是leaf节点,是否满足target==0条件
如果找到了满足条件的leaf节点,怎么把完整路径保存,这个我是没有想到的
ArrayList<ArrayList<Integer>> path = new ArrayList<ArrayList<Integer>>();
定义为类的属性,提升作用域
然后这种ArrayList的初始化方法,就是把temp进行了一个拷贝
path.add(new ArrayList<Integer>(temp));
对于我这个菜鸡来说,真是米奇老鼠到了妙妙屋,妙到家了
往前回溯时需要把当前保存的节点都删除了
temp.remove(temp.size()-1);就有了这个操作
public class FindPath {
ArrayList<ArrayList<Integer>> path = new ArrayList<ArrayList<Integer>>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
test(new ArrayList<Integer>(),root,target);
return path;
}
//拿着这个target一直往下减节点对应的值,如果到了根节点为0了说明这条路径符合条件
public void test(ArrayList<Integer> temp, TreeNode root, int target){
if(root==null) return;
temp.add(root.val);
target=target- root.val;
if(root.right==null&&root.left==null&&target==0) {
path.add(new ArrayList<Integer>(temp));
}
test(temp,root.left,target);
test(temp,root.right,target);
temp.remove(temp.size()-1);
}
}