ListToTree 列表转树 简单方案,

同事最近做了一个列表转树,我提供了方法,但是他使用的是对象,不是json,所以做的时候很为难,json转来转去很难受,所以继续优化一版出来,参考如下

1:简单对象如下

public class Dept {

    private String code;
    private String parentCode;
    public List<Dept> children;
    /*其他字段*/
    private String name;
//注意addChild 和下面的MerageChild方法,需要手动添加一下 public Dept addChild(Dept dept){ children = children==null?new ArrayList<Dept>():children; children.add(dept); return this; } public Dept merageChild(Dept dept){ children = children==null?new ArrayList<Dept>():children; if(dept!=null) children.addAll(dept.children); return this; } public String getCode() { return code; } public void setCode(String code) { this.code = code; } public String getParentCode() { return parentCode; } public void setParentCode(String parentCode) { this.parentCode = parentCode; } public List<Dept> getChildren() { return children; } public void setChildren(List<Dept> children) { this.children = children; } public String getName() { return name; } public void setName(String name) { this.name = name; } }

  上面的对象新增了addChild 和merageChild方法

    public static List<Dept> listToTree(List<Dept> list){
        HashMap<String,Dept> cache = new HashMap<String,Dept>(list.size());
        List<Dept> root = new ArrayList();
        for(Dept dept:list){
            String parentCode = dept.getParentCode();
            cache.merge(dept.getCode(), dept,(ov,nv)->nv.merageChild(ov));
            if(ObjectUtils.isEmpty(parentCode)){
                root.add(dept);
            }else{
                //会生成一个新的临时对象,暂存未处理到的对象
                cache.compute(parentCode,(k,v)->v==null?new Dept().addChild(dept):v.addChild(dept));
            }
        }
        return root;
    }

 

列表转树方法如上,总共不到10行代码,还是比较简单的,这个方案中,没有频繁的创建对象,这是比上一次的优化(还是会创建,体现在先遍历到子对象的时候)。

posted @ 2021-08-28 07:10  anscript  阅读(404)  评论(0编辑  收藏  举报