java 树形结构对象生成

最新推荐文章于 2024-07-28 18:44:06 发布

永恒之夜

最新推荐文章于 2024-07-28 18:44:06 发布

阅读量769

点赞数

CC 4.0 BY-SA版权

分类专栏： java

本文链接：https://blog.youkuaiyun.com/qq_35589080/article/details/114987026

java 专栏收录该内容

4 篇文章

订阅专栏

本文介绍了一种高效的树形结构生成算法，通过使用HashMap快速查找特性，将组织结构数据转换为树形结构，有效提高了生成速度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

常规的树形结构生成通常使用递归的方式进行，以深度遍历的方式不断将子节点追加到父节点上，在效率上并不可观。记录一种生成树形结构的算法。

代码如下：

   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中找到对应的根节点，即可获取一个树或树林。