题目:输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路:先找到二叉树的所有路径,然后计算和,取符合标准的路径:
public class Solution {
ArrayList<ArrayList<Integer>> result = new ArrayList();//所有路径
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
if(root!=null){
ArrayList<Integer> rootList = new ArrayList();
result.add(rootList);
Traverse(root,rootList);
}
ArrayList<ArrayList<Integer>> realResult = new ArrayList();
for(ArrayList<Integer> list:result){
int count = 0;
for(int val:list){
count+=val;
}
if(count == target){
realResult.add(list);
}
}
return realResult;
}
public void Traverse(TreeNode root,ArrayList<Integer> list){
ArrayList<Integer> rootList = list;
rootList.add(root.val);
if(root.left!=null){
ArrayList<Integer> templ = new ArrayList();
result.remove(rootList);
result.add(templ);
templ.addAll(rootList);
Traverse(root.left,templ);
}
if(root.right!=null){
ArrayList<Integer> tempr = new ArrayList();
result.remove(rootList);
result.add(tempr);
tempr.addAll(rootList);
Traverse(root.right,tempr);
}
}
}