题目
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路
使用先序遍历(最先访问到根节点的值),递归每个树节点的左右子树,如果该节点为叶子节点且和为给定的值,则将该路径加入目标集;否则回退当前路径到上一状态,继续访问左右子树。
代码
public class Solution {
private ArrayList<ArrayList<Integer>> ret = new ArrayList<ArrayList<Integer>>();
private ArrayList<Integer> list = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root == null) return ret;
target = target - root.val;
list.add(root.val);
if(target == 0 && root.left == null && root.right == null){
ret.add(new ArrayList<Integer>(list));
}else{
FindPath(root.left, target);
FindPath(root.right, target);
}
list.remove(list.size() - 1);
可以加上也可以不加
//target = target + root.val;
return ret;
}
}
其中,//target = target + root.val; 这行代码纠结了一会,为什么list需要在每轮递归返回时把最后一个元素去掉,而target变量不需要在每轮递归返回时加上 此轮减去的元素?
那是因为每轮递归时会维护一个递归栈,递归栈在每次递归时都会保存当前方法中的参数状态,因为target是方法的入参,所以在每轮递归时会保存当前的状态,而list是全局变量,状态并不在递归栈中保存,需要每次返回时都去掉最后一个元素。所以即使加上注释那一行代码,target的值也仅影响当前递归栈中的target值。