剑指offer-二叉树中和为某一值的路径

题目描述

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

 

使用dfs

public class Solution {//dfs 树 my
    ArrayList<ArrayList<Integer>> re ;
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        re = new ArrayList<>();
        if(root!=null){
            dfs(root,new ArrayList<Integer>(),target,0);
        }
        return re;
    }
    public void dfs(TreeNode root ,ArrayList<Integer> list,int target,int sum){
        if(root.left==null&&root.right==null){
            if((sum+root.val)==target){
                //添加路径
                ArrayList l =new ArrayList<Integer>(list);
                l.add(root.val);
                re.add(l);
                return;
            }
            return;
        }
        sum +=root.val;
        list.add(root.val);
        if(root.left!=null){
            dfs(root.left,list,target,sum);
        }
        if(root.right!=null){
            dfs(root.right,list,target,sum);
        }
        sum-=root.val;
        list.remove(list.size()-1);
        return ;
    }
}

其中和可简化为一个参数，如下

public class Solution {
    ArrayList<ArrayList<Integer>> re ;
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        re = new ArrayList<>();
        if(root!=null){
            dfs(root,new ArrayList<Integer>(),target);
        }
        return re;
    }
    public void dfs(TreeNode root ,ArrayList<Integer> list,int target){
        if(root.left==null&&root.right==null){
            if(root.val==target){
                //添加路径
                ArrayList l =new ArrayList<Integer>(list);
                l.add(root.val);
                re.add(l);
                return;
            }
            return;
        }
        list.add(root.val);
        if(root.left!=null){
            dfs(root.left,list,target-root.val);
        }
        if(root.right!=null){
            dfs(root.right,list,target-root.val);
        }
        list.remove(list.size()-1);
        return ;
    }
}