欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

List转tree的方法,支持数据量很大的情况(效率对比)

程序员文章站 2022-07-13 08:06:10
...

两种方式对比:
  数据库中有两个字段,分别代表本数据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秒。

相关标签: 技术总结