输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。
路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
import java.util.ArrayList;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
private ArrayList<ArrayList<Integer>> cap = new ArrayList<ArrayList<Integer>>();
private ArrayList<Integer> road = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root ==null || target ==0){
return cap;
}
road.add(root.val);
if(root.left == null && root.right == null){ //如果加到叶节点,那么就把路径中所有的值加起来判断是否与目标相等
int len = 0;
for(int i =0; i< road.size();i++){
len +=road.get(i);
}
if(len == target){
cap.add(new ArrayList<Integer>(road));
}
}else{ //如果不是叶节点,就递归,先遍历完左节点,再遍历右节点
FindPath(root.left,target);
FindPath(root.right,target);
}
road.remove(new Integer(root.val)); //路径每次遍历一个节点就要删除一个,向上一层父节点移动
//remove有方法重载,remove(int index),remove(Object o),如果直接remove(root.val)会认为是移除下标为root.val的对象,可能报边界溢出异常
return cap;
}
}