题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
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>> ans;
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
ans = new ArrayList<ArrayList<Integer>>();
ArrayList<Integer> list= new ArrayList<>();
solve(root,target,0,list);
change();
return ans;
}
private void solve(TreeNode node,int target,int sum,ArrayList<Integer> list){
if(node !=null){
sum+=node.val;
list.add(node.val);
if(node.left==null && node.right==null){
if(sum==target){
ArrayList<Integer> res=new ArrayList<>(list);
ans.add(res);
}
}else{
solve(node.left,target,sum,list);
solve(node.right,target,sum,list);
}
//消除当前节点对查询路径的影响
list.remove(list.size()-1);
}
}
//选择排序
private void change(){
for(int i=0;i<ans.size();i++){
int index=i;
for(int j=i+1;j<ans.size();j++){
if(ans.get(j).size()>ans.get(index).size()){
index=j;
}
}
if(i!=index){
ArrayList<Integer> temp=ans.get(i);
ans.set(i,ans.get(index));
ans.set(index,temp);
}
}
}
}