题目:输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路:
递归先序遍历树, 把结点加入路径。
若该结点是叶子结点则比较当前路径和是否等于期待和。
弹出结点,每一轮递归返回到父结点时,当前路径也应该回退一个结点
Java代码
import java.util.ArrayList;
/*
//TreeNode也可以写成内部类
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>> arrlist = new ArrayList<ArrayList<Integer>>();
if(root == null) return arrlist;
ArrayList<Integer> a1 = new ArrayList<Integer>();
int sum = 0;
sums(root, target, arrlist,a1, sum);
return arrlist;
}
public void sums(TreeNode root, int target, ArrayList<ArrayList<Integer>> arrlist, ArrayList<Integer> a1, int sum){
if(root == null) return;
sum = sum + root.val;
if(root.left == null && root.right == null){
if(sum == target){
a1.add(root.val);
arrlist.add(new ArrayList<Integer>(a1));
//
a1.remove(a1.size()-1);
}
return;
}
a1.add(root.val);
sums(root.left, target, arrlist,a1, sum);
sums(root.right, target, arrlist,a1, sum);
a1.remove(a1.size()-1);
}
}
更简洁的写法
// super nice
public class Solution{
private ArrayList<ArrayList<Integer>> listAll = new ArrayList<ArrayList<Integer>();
private ArrayList<Integer> list = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPaths(TreeNode root,int target) {
if(root == null) return listAll;
list.add(root.val);
target -= root.val;
if(target == 0 && root.left == null && root.right == null)
listAll.add(new ArrayList<Integer>(list));
FindPath(root.left, target);
FindPath(root.right, target);
list.remove(list.size()-1);
return listAll;
}
//构建一个 TreeNode 内部类
class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}