剑指offer(二十四)——二叉树中和为某一值的路径
题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
题解
这道题难度一般,唯一需要注意的是数组长度大的要在前面,这个问题还是比较好解决的,只需要对结果数组进行排序就可以了。但在测试用例中,没有实现排序也可以过,这是它不严谨的地方了。
下面给出我的方法,递归深搜就可以了。
Collections.sort(ArrayList<>(),new Comparator<T>(){});
使用Collections的排序方法,通过Comparator的匿名内部类自定义排序方式,具体使用方法自行百度。- 还有一个坑点:关于ArrayList.add()方法的,add()方法添加的是引用地址,在使用该方法时,最好只用于添加数字,假如要添加其他类型变量时,最好当场new一个。
- 理解一下上面这个坑点,比如
ArrayList<Integer> list1 = new ArrayList<>();
ArrayList<ArrayList<Integer>> list2 = new ArrayList<>();
//假如你直接
list2.add(list1);
//那么假如后面list1变了,那list2中的元素也会跟着变,更别说假如list1是一个局部变量,list2是全局变量的情况。
//所以一般我们用
list2.add(new ArrayList<>(list));
//new一个新的ArrayList就不会受到list1的影响了
public static ArrayList<ArrayList<Integer>> resultList = new ArrayList<>();
public static ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
ArrayList<Integer> list = new ArrayList<>();
Search(root, target, 0, list);
//自定义排序方法
Collections.sort(resultList, new Comparator<ArrayList<Integer>>() {
@Override
public int compare(ArrayList<Integer> o1, ArrayList<Integer> o2) {
// TODO 自动生成的方法存根
if (o1.size() < o2.size()) {
return 1;
}else {
return -1;
}
}
});
return resultList;
}
public static void Search(TreeNode root, int target, int count, ArrayList<Integer> list) {
if (root == null) {
return;
}
int temptarget = count + root.val;
list.add(root.val);
if (root.left == null && root.right == null && temptarget == target) {
//坑点:resultList.add(list);会报错
resultList.add(new ArrayList<>(list));
}else {
if (root.left != null) {
Search(root.left, target, temptarget, list);
}
if (root.right != null) {
Search(root.right, target, temptarget, list);
}
}
list.remove(list.size() - 1);
}