题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路分析
递归遍历树, 把结点加入路径。若该结点是叶子结点且当前路径和等于期待和,则将此路径加入返回的lists中,否则退回父结点。在每一轮递归返回到父结点时,当前路径也要回溯一个结点值。
代码实现
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
ArrayList<ArrayList<Integer>> arrayLists = new ArrayList<>();
if (root == null || root.val > target) {
return arrayLists;
}
arrayLists = process(root, target, arrayLists, new ArrayList<Integer>());
return arrayLists;
}
public ArrayList<ArrayList<Integer>> process(TreeNode root, int target, ArrayList<ArrayList<Integer>> arrayLists, ArrayList<Integer> list) {
if(root==null){
return arrayLists;
}
target -= root.val;
list.add(root.val);
if (target == 0 && root.right == null && root.left == null) {
arrayLists.add(new ArrayList<>(list));
}
process(root.left, target, arrayLists, list);
process(root.right, target, arrayLists, list);
list.remove(list.size() - 1);
return arrayLists;
}