题目描述
输入一颗二叉树和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。
模版:
public class Solution {
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
}
}
树的问题一定要递归:所以代码中
FindPath(TreeNode root.left,int target);
FindPath(TreeNode root.right,int target);
是必须的。
而且想到想获得正好和为target的路径是很难的,所以应该获得所有路径(即把所有遍历到的点都压入Llst)然后把正好是的加入ListAll(即所有路径的集合)。
那什么时候是呢?每次压入一个target都会减少root.val,当减完正好为0且为叶结点就是了。
然后继续向下走,这时看target是否大于0,因为如果小于0没必要继续向下了。
最后一定别忘了移除末端节点,因为无论是不是目标路径,这条路径都结束了,那么当然必须删除这个节点,不然list含有这个节点没法表示其他路径。每个节点在左右子树都遍历完毕后都会删去,因为这时包含这个节点的所有路径都已经判断过了。
完整代码:
public
class
Solution {
private
ArrayList<ArrayList<Integer>> listAll =
new
ArrayList<ArrayList<Integer>>();
private
ArrayList<Integer> list =
new
ArrayList<Integer>();
public
ArrayList<ArrayList<Integer>> FindPath(TreeNode root,
int
target) {
if
(root ==
null
)
return
listAll;
list.add(root.val);//list的添加和删除两句是这段代码的关键
target -= root.val;
if
(target ==
0
&& root.left ==
null
&& root.right ==
null
)
listAll.add(
new
ArrayList<Integer>(list));
else if(target > 0){
FindPath(root.left, target);
FindPath(root.right, target);
}
list.remove(list.size()-
1
);//必须在list里删去当前节点,因为执行到这里就结束了(无论这条路径是不是目标路径),必须删去list才能表示其他路径
return
listAll;
}
}