题目:
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路:
最早的思路是想通过栈来实现,但是这道题每条路径都属于一个新的链表.所以这种实现方法会显得冗余.
通过递归的方式进行实现,目标的值可以通过一直递减然后直到为0,且为叶子节点来判断是否为合适的路径.
代码
public class FindPath {
private ArrayList<ArrayList<Integer>> arrayLists = new ArrayList<>();
private ArrayList<Integer> arrayList = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
if(root == null){
return arrayLists;
}
arrayList.add(root.val);
target -= root.val;
//target<0说明不需要在遍历了,直接返回
if(target < 0){
arrayList.remove(arrayList.size() - 1);
return arrayLists;
}
if(target == 0 && root.left == null && root.right == null)
arrayLists.add(new ArrayList<Integer>(arrayList));
if(root.left != null)
FindPath(root.left,target);
if(root.right != null)
FindPath(root.right,target);
//回档
arrayList.remove(arrayList.size()-1);
return arrayLists;
}
}