二叉树中和为某一值的路径
题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
分析
使用递归,每次找到符合的路径后进行回退继续寻找。
实现
private ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
private ArrayList<Integer> list = new ArrayList<>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root, int target) {
if(root == null)
return result;
list.add(root.val);
// 加入当前节点,减去节点值
target -= root.val;
if(target == 0 && root.left == null && root.right == null)
// 如果target = 0,说明满足条件
result.add(new ArrayList<Integer>(list));
// 分别搜索左右子树
FindPath(root.left, target);
FindPath(root.right, target);
// 因为每一个结果集(路径)使用的是同一个list,所以回退一步寻找其他路径
list.remove(list.size()-1);
return result;
}