常规的树形结构生成通常使用递归的方式进行,以深度遍历的方式不断将子节点追加到父节点上,在效率上并不可观。记录一种生成树形结构的算法。
代码如下:
public List<OrganizationVo> getStateTree() {
List<OrganizationVo> orgs = organizationMapper.getStateOrg();
//将组织数据转换为已组织ID为key的Map
Map<String, OrganizationVo> nodeMap = orgs.stream().
collect(Collectors.toMap(OrganizationVo::getId, Function.identity()));
//根节点容器;
Set<String> parentSet=new HashSet<>();
nodeMap.forEach((id, node) -> {
//从nodeMap中寻找父节点
String parentId = node.getParentId();
OrganizationVo parent = nodeMap.get(parentId);
if (parent == null) {
//如果当前节点没有父节点(认为其是父节点),将其存入parentSet
parentSet.add(id);
} else {
//否则,将当前节点存入父节点,父节点回归nodeMap
if (parent.getChildren() == null) {
List<OrganizationVo> list = new ArrayList<>();
parent.setChildren(list);
}
parent.getChildren().add(node);
nodeMap.put(parentId, parent);
}
});
//遍历parentSet 从nodeMap中取出树结构
List<OrganizationVo> rs=new ArrayList<>();
parentSet.forEach(id->{
OrganizationVo node=nodeMap.get(id);
if(node!=null){
rs.add(node);
}
});
return rs;
}
该算法将一组树形结构的对象取出在一个map中,利用HashMap查询快的优势,缩减处理时间。
遍历Map,对,找出其中的根节点的标识记录在缓存区,对不是根节点的节点则寻找其父节点建立关联。
最后从缓存区取出根节点标识,从map中找到对应的根节点,即可获取一个树或树林。