题目描述:
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
分析:
整体思路:
定义一个要返回的指定类型(ArrayList<ArrayList<Integer>>)的对象result,在定义一个ArrayList<Integer>类型的对象list,list的作用是为了遍历时记录节点,然后找到符合条件的节点路径,最后直接调用result 的 add 方法把符合条件的 list 添加到 result 中。
找路径的具体思路:
1. 判断给定的二叉树是否为空,如果根节点都为空的话,那么就直接返回null。
2.如果不为空,将根节点放入list中,用当前的target减去当前根节点的val值来作为新的target值,利用的是减法来计算,直到target减为零并且当前节点必须没有左右孩子(即叶子节点),此时就将list添加到result中,再将result返回。如果一直遍历到叶子节点,target都没有刚好为零,那么就还是返回result,但是result中是没有值的。
代码:
import java.util.*;
/**
public class TreeNode {
int val = 0;
TreeNode left = null;
TreeNode right = null;
public TreeNode(int val) {
this.val = val;
}
}
*/
public class Solution {
ArrayList<ArrayList<Integer>> result = new ArrayList<ArrayList<Integer>>();
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;
}
}