题目描述
输入一棵二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径;
路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径;
解题思路
1.对其进行先序遍历,将保存当前节点;
2.每当遍历到叶子节点的时候,检查和是否等于给定的值;
3.每当要返回到当前节点的父节点的时候,要弹出当前元素;
private ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>>(); //用来保存所有符合要求的路径
private ArrayList<Integer> list = new ArrayList<Integer>(); //用来保存当前路径
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
if (root == null){
return listAll;
}
list.add(root.val); //将当前节点加入路径
target = target - root.val;
if (target == 0 && root.left == null && root.right == null){
//如果当前路径的和等于target,并且它是叶子节点
//这说明,当前路径是符合要求的,就将这条路径加入listAll
//需要注意的是,必须new ArrayList<Integer>(list),不能直接添加list
listAll.add(new ArrayList<Integer>(list));
}
FindPath(root.left, target);
FindPath(root.right, target);
list.remove(list.size()-1); //返回到父结点的时候,必须把当前节点从路径中删除
return listAll;
}