题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路分析
先序遍历,访问节点时,把该节点添加到路径上去,并累加该节点的值
如果该节点为叶节点,并且路径中节点值的和刚好等于输入的整数,则符合当前路径要求
当前节点访问完毕,要回到父节点访问其他节点,所以,在退出方法前要在路径上删除当前节点并减去当前节点值,以确保返回父节点时路径刚好是从根节点到父节点。
代码
ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> list1 = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if (root == null) { //边界条件
return list;
}
list1.add(root.val); // 添加根节点的值
target = target - root.val;
if (target==0 && root.left==null && root.right==null) { //形成路径且和为target,则添加入list
list.add(new ArrayList<Integer>(list1));
}
FindPath(root.left, target); //对左右子树开始递归,即遍历整个树
FindPath(root.right, target);
list1.remove(list1.size()-1);
return list;
}
结果