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