题目

  • 输入一颗二叉树的跟节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中，数组长度大的数组靠前)

思路


代码

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 {
    //存储路径的容器
    ArrayList<ArrayList<Integer>> resultlist = new ArrayList<>();
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        //每个路径----用来存储节点的容器
        ArrayList<Integer> path = new ArrayList<Integer>();
        if(root==null)
            return resultlist;
        //路径每个节点数值之和
        int currentSum =0;
        FindPath(root,target,currentSum,path);
        return resultlist;
    }
    public void FindPath(TreeNode root,int target,int currentSum,ArrayList<Integer> path){
        if(root==null)
            return;
        //首先判断第一个根节点的值与target是否相同
        currentSum+=root.val;
        //将根节点元素添加到路径中
        path.add(root.val);
        //如果根节点没有左右节点，那么则直接判断是否值为target，如果相同则添加路径path到resultlist中
        if(root.left==null && root.right ==null){
            if(currentSum==target){
                resultlist.add(new ArrayList<Integer>(path));
                //将路径的叶节点删除，再继续寻找别的情况target
                path.remove(path.size()-1);
                return;
            }
        }
        //如果根节点还有左右子节点，那么直接递归左右子树的情况
        FindPath(root.left,target,currentSum,path);
        FindPath(root.right,target,currentSum,path);
        //这里左右子树都找完了，回到了找完的左右子树的父节点
       //由于父节点的左右子树找完了，所以父节点这里也没有用了，把父节点删除
        path.remove(path.size()-1);
    }
}