输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。
路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
思路:
1、从根节点开始访问二叉树,使用前序遍历
2、累积从根节点开始访问的值的和,并将节点的值加入inList,若累积值与寻找的值相等,将inList加入返回的outList
3、当累积值超过寻找值时,返回上一节点,此时,inList中的末尾元素要删除
import java.util.ArrayList;
import java.util.Iterator;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
ArrayList<ArrayList<Integer>> outList = new ArrayList<ArrayList<Integer>>();
if(null != root){
ArrayList<Integer> inList = new ArrayList<Integer>();
findPath(root,0,target,inList,outList);
}
return outList;
}
public void findPath(TreeNode root,int curSum,int target,ArrayList<Integer> inList,
ArrayList<ArrayList<Integer>> outList){
if(null !=root){
curSum += root.val;
inList.add(root.val);
if(curSum < target){
findPath(root.left,curSum,target,inList,outList);
findPath(root.right,curSum,target,inList,outList);
}
if(curSum == target){
if(null == root.left && null == root.right){
ArrayList<Integer> addList = new ArrayList<Integer>();
Integer value;
Iterator it = inList.iterator();
while(it.hasNext()){
value = (Integer)it.next();
addList.add(value);
}
outList.add(addList);
}
}
inList.remove(inList.size()-1);
}
}
}