题目
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
分析
思路一:
路径:从根节点开始
根据树的路径的定义,我们只能采用前序遍历。而又由于题目给的节点定义中为给出其父节点的指向,所以只能遍历的时候将遍历到的节点值一一存入一个“容器”并将该节点值累加。直到遍历到叶节点,如果当前值累加的和等于输入的整数,则存储该“容器”(路径)。如果不等于则需要返回到父节点再遍历另一子节点,这样我们就需要将“容器”当前的最后一个值删掉然后再存储。可以发现这种容器符合栈的“后进先出”规则。而我们采用递归实现这种“容器”的值的添加与删除,正好符合“后进先出”。
时间复杂度:o(N)
空间复杂度:o(N)
代码:
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>> result = new ArrayList<ArrayList<Integer>>();
private ArrayList<Integer> list = new ArrayList<Integer>();
public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
//递归头
if(root==null) return result;
//递归体
list.add(root.val);
target -= root.val;
if(target==0&&root.left==null&&root.right==null) result.add(new ArrayList<Integer>(list));
FindPath(root.left, target);
FindPath(root.right, target);
list.remove(list.size()-1);
return result;
}
}