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行代码,还是比较简单的,这个方案中,没有频繁的创建对象,这是比上一次的优化(还是会创建,体现在先遍历到子对象的时候)。

浙公网安备 33010602011771号