二叉树中和为某一值的路径(dfs,二维ArrayList及其排序)

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

思路:dfs

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 {
    public ArrayList<ArrayList<Integer>> FindPath(TreeNode root,int target) {
        ArrayList<ArrayList<Integer>> ans=new ArrayList<ArrayList<Integer>>();
        ArrayList<Integer> list=new ArrayList<Integer>();
        dfs(root,target,list,ans);
        Collections.sort(ans,new Comparator<ArrayList<Integer>>(){//对数组排序
            public int compare(ArrayList<Integer> a,ArrayList<Integer> b){
                return a.size()<b.size()?1:-1;
            }            
        });
        return ans;
    }
    public void dfs(TreeNode root,int target,ArrayList<Integer> arr,ArrayList<ArrayList<Integer>> ans){
        if(root==null) return ;
        int val=root.val;
        if(val>target) return ;
        arr.add(val);
        target-=val;
        if(target==0&&root.left==null&&root.right==null){
            ans.add(new ArrayList<Integer>(arr));//一定要new
        }
        else{
            dfs(root.left,target,arr,ans);
            dfs(root.right,target,arr,ans);
        }
        arr.remove(arr.size()-1);
    }
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值