List转tree的方法,支持数据量很大的情况(效率对比)
两种方式对比:
数据库中有两个字段,分别代表本数据id,和父数据id,还有一个所需数量的字段。如果有一条数据的父数据id为null,说明这是*节点。我的需求是把这种关系的数据转成树形结构,
第一种方法是双重for循环(不支持数据量大的情况):
private static List listGetStree(List list) {
List treeList = new ArrayList();
for (TreeNode tree : list) {
//找到根
if (tree.getPid() == 0) {
treeList.add(tree);
}
//找到子
for (TreeNode treeNode : list) {
if (treeNode.getPid() == tree.getId()) {
if (tree.getChildren() == null) {
tree.setChildren(new ArrayList());
}
tree.getChildren().add(treeNode);
}
}
}
return treeList;
}
由于我的代码复杂,所以放简单的(别人的)。外层for循环在找根,是把所有数据都当成根,第二个for循环就是在找子,帮所有的根找子,找到子了,我们就放到TreeNode类里面的一个List 里面。(这里注意要有这么一个类。推荐链接:https://blog.csdn.net/jcroad/article/details/79735790)
第二种方法才是重点
两个for循环(不是双重的,支持数据量大的情况)
Map<Long, DepartmentVO> departmentMap = new HashMap<>();
List departmentVOS = new ArrayList<>();
departments.forEach(e -> {
DepartmentVO departmentVO = new DepartmentVO();
BeanUtils.copyProperties(e, departmentVO);
departmentMap.put(e.getDepartmentId(), departmentVO);
});
for (Department department : departments) {
DepartmentVO departmentVOparent = departmentMap.get(department.getParent());//根据父id去拿构建的树
DepartmentVO departmentVO = departmentMap.get(department.getDepartmentId());//根据子id去拿构建的树
if (departmentVOparent == null) {//为空了说明根据父id去拿没拿到,那么父id为null,本次循环找到了*
departmentVOS.add(departmentVO);//那么就把这个*数据的树放到list里面存起来
continue;
}
departmentVOparent.getSubDepartment().add(departmentVO);//如果不是*数据,我们就把本数据id构建的树,放到父id构建树的里面的list里面。这样我们就可以把所有的数据全部组装成父子关系
}
由于各自需求不同,我就说一下思路:第一个for循环就是把所有的数据用自己的本数据id构建一个类(树)。第二个循环是在组装父子关系,拿到子数据和父数据,把子数据放进父数据里的list里面。
对比:双重for循环循环次数是肯定要比两个循环次数多。我在开发种就遇见list数据多达2000多条(上一篇动态sql博客),刚开始没注意,使用第一种方式,结果跑循环平均时间是0.8秒,而使用两个for循环平均时间是0.15秒。
上一篇: es6 var 与 let 区别
下一篇: 记一次数据库死锁追踪过程