java 树结构分级

温和的洛瑞

已于 2022-09-20 15:39:17 修改

阅读量656

点赞数 3

分类专栏： Java 文章标签： java

于 2021-10-18 15:19:57 首次发布

本文链接：https://blog.csdn.net/qq_41759247/article/details/120826689

版权

Java 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

这篇博客介绍了如何使用栈数据结构配合HashMap来遍历和组织具有层级关系的树形数据，如项目DTO列表。通过不断入栈和出栈子节点，将同一层级的项目Dto归为一组，并存储到childrenMap中，实现层级数据的整理。整个过程确保了数据的正确组织，适用于多级嵌套的数据结构处理。

摘要由CSDN通过智能技术生成

树结构数据

代码

        Stack<List<ProjectDto>> listStack = new Stack<>();

        // 存储层级数据
        Map<Integer, List<ProjectDto>> childrenMap = new HashMap<>();

        // 入栈
        listStack.push(projectDtoList1);

        // 层级号
        int count = 0;

        // 项目不为空时
        while (!listStack.isEmpty()) {

            // 出栈
            List<ProjectDto> projectDtoList2 = listStack.pop();

            // 存储当前层级下的子节点
            List<ProjectDto> projectDtoArrayList3 = new ArrayList<>();

            for (int i = 0; i < projectDtoList2.size(); i++) {
                if (!projectDtoList2.get(i).getParentId().equals("0")) {
                    if (projectDtoList2.get(i) != null) {
                        // 添加所有的子节点
                        projectDtoArrayList3.addAll(projectDtoList2.get(i).getChildren());
                    }
                } else {
                    projectDtoArrayList3.addAll(projectDtoList2.get(i).getChildren());
                }

                // 循环完毕，添加当前节点数据
                if (i == projectDtoList2.size() - 1) {
                    childrenMap.put(count++, projectDtoList2);
                }

            }
            if (projectDtoArrayList3.size() != 0 && projectDtoArrayList3 != null) {
                // 获取子节点入栈
                listStack.push(projectDtoArrayList3);
            }
        }
        map.put("Level", childrenMap);