输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
package ds;
import java.util.ArrayList;
public class FindPathToTarget {
private static class TreeNode {
int val;
TreeNode left;
TreeNode right;
public TreeNode(int val) {
this.val = val;
}
}
private static ArrayList<ArrayList<Integer>> result = new ArrayList<>();
private static ArrayList<Integer> temp = new ArrayList<>();//为了防止在多次递归中被覆写,需要写在外面
public static void main(String[] args) {
TreeNode root = new TreeNode(6);
TreeNode node11 = new TreeNode(2);
TreeNode node12 = new TreeNode(3);
TreeNode node21 = new TreeNode(1);
TreeNode node22 = new TreeNode(5);
TreeNode node31 = new TreeNode(3);
TreeNode node42 = new TreeNode(4);
root.left = node11;
root.right = node12;
node11.left = node21;
node11.right = node22;
node22.left = node31;
node31.right = node42;
findPath(root,9);
System.out.println(result);
}
public static ArrayList<ArrayList<Integer>> findPath(TreeNode root, int target) {
if (root == null) {
return result;
}
temp.add(root.val);
target = target - root.val;
if (target < 0) {
temp.remove(temp.size() - 1);
return result;
}
if (target == 0 && root.left == null && root.right == null) {
result.add(new ArrayList<>(temp));
//不能直接add(temp),因为传递的是引用的值,也就是temp所指对象的内存地址
//当temp所指对象的内容改动时,result里面的对应位置上的元素所指的内容当然也会改动,因为是同一个地址
//而新建一个对象只是利用了temp所指对象内存里面存储的值,没有建立地址联系,当temp之后再改变时新对象里面的值不会改变
}
if (root.left != null) {
findPath(root.left, target);
}
if (root.right != null) {
findPath(root.right, target);
}
//其余情况:target>0并且root.left==null&&root.right==null,回归到上一层去右子树里面找
temp.remove(temp.size() - 1);
//回到上一层时,不需要再把target+=root.val,因为上一层的target是上上一层调用函数时压进栈里面临时存储的,
//直接弹出来用,在外部对target进行操作不会影响栈里面的值
return result;
}
}