题目:输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路:思路很简单,实现很困难,嗯,对实现很困难。
下面我来说一下思路,
有几个定义提前说一下,输入的整数定义为int target,节点定义为TreeNode root,用来存放一条路径的集合ArrayList<Integer> list,返回结果ArrayList<ArrayList<Integer>> result;
我们使用前序遍历的方法,如果每次走过的节点的值root.val<target就把这个节点的值存到list集合中,并设target-=root.val,如果target==0并且root为叶子节点,那么就把list集合存到result中。
代码:
ArrayList<ArrayList<Integer>> list = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> temp = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root==null)
return list;
target -=root.data;
temp.add(root.data);
if(target ==0&&root.left==null&&root.right==null){
//创建一个新的空间存放一条路径的值,避免后面remove操作影响结果
list.add(new ArrayList<Integer>(temp));
}else
{
FindPath(root.left,target);
FindPath(root.right,target);
}
//每算出来一条路径,并且放入result后,就将list最后一个元素删除
temp.remove(temp.size()-1);
return list;
}