JAVA数据结构

树形结构

java基于Map实现集合组装树形结构

public class TestMain {@Data
//    @Builder
//    @NoArgsConstructor@AllArgsConstructorpublic static class DeptEntity {private   Long deptId;private   Long parentId;private   String deptName;private   List<DeptEntity> children = new ArrayList<>();public DeptEntity(Long deptId, Long parentId, String deptName) {this.deptId = deptId;this.parentId = parentId;this.deptName = deptName;}public void addChild(DeptEntity child) {this.children.add(child);}}public static List<DeptEntity> getData() {List<DeptEntity> list = new ArrayList<>();
//        list.add(new DeptEntity(100L, 0L, "若依科技"));list.add(new DeptEntity(101L, 100L, "深圳总公司"));list.add(new DeptEntity(102L, 100L, "长沙分公司"));list.add(new DeptEntity(103L, 101L, "研发部门"));list.add(new DeptEntity(104L, 101L, "市场部门"));list.add(new DeptEntity(105L, 102L, "运维部门"));list.add(new DeptEntity(106L, 102L, "财务部门"));return list;}public static void mapBuildTree() {// 获取假数据List<DeptEntity> data = getData();List<DeptEntity> roots = new ArrayList<>();Map<Long, DeptEntity> childMap = new HashMap<>();// 1.将所有节点存入Map中for (DeptEntity deptEntity : data) {childMap.put(deptEntity.getDeptId(), deptEntity);}// 2.构建树结构for (DeptEntity deptEntity : data) {// 根节点if (deptEntity.getParentId() == null || deptEntity.getParentId() == 0 || !childMap.containsKey(deptEntity.getParentId())) {roots.add(deptEntity);} else {// 非根节点，加入到父节点的children列表DeptEntity parent = childMap.get(deptEntity.getParentId());if (parent != null) {parent.addChild(deptEntity);}}}System.out.println(JSONObject.toJSONString(roots));}public static void main(String[] args) {mapBuildTree();}}

引用 https://blog.csdn.net/zhangfuping123456789/article/details/146275943
引用2：https://www.jb51.net/program/339751h5u.htm

java基于Stream实现集合组装树形结构

public class TestMain {@Data
//    @Builder
//    @NoArgsConstructor@AllArgsConstructorpublic static class DeptEntity {private   Long deptId;private   Long parentId;private   String deptName;private   List<DeptEntity> children = new ArrayList<>();/*** 排序*/private Integer sort;public DeptEntity(Long deptId, Long parentId, String deptName) {this.deptId = deptId;this.parentId = parentId;this.deptName = deptName;}public void addChild(DeptEntity child) {this.children.add(child);}}public static List<DeptEntity> getData() {List<DeptEntity> list = new ArrayList<>();
//        list.add(new DeptEntity(100L, 0L, "若依科技"));list.add(new DeptEntity(101L, 100L, "深圳总公司"));list.add(new DeptEntity(102L, 100L, "长沙分公司"));list.add(new DeptEntity(103L, 101L, "研发部门"));list.add(new DeptEntity(104L, 101L, "市场部门"));list.add(new DeptEntity(105L, 102L, "运维部门"));list.add(new DeptEntity(106L, 102L, "财务部门"));return list;}public static void main(String[] args) {List<DeptEntity> treedata =  listWithTree();System.out.println(JSONObject.toJSONString(treedata));}// 构建树形结构的分类列表public static List<DeptEntity> listWithTree() {// 从某个静态源（可能是数据库或配置）获取所有分类的列表List<DeptEntity> categories = getData();Map<Long, DeptEntity> childMap = new HashMap<>();// 1.将所有节点存入Map中for (DeptEntity deptEntity : categories) {childMap.put(deptEntity.getDeptId(), deptEntity);}// 过滤出所有一级分类（即父分类ID为null的分类），并为每个一级分类设置子分类List<DeptEntity> collect = categories.stream() // 将列表转换为流.filter(m -> !childMap.containsKey(m.getParentId())) // 过滤出一级分类.map((m) -> { // 对每个一级分类进行处理m.setChildren(getChildrenList(m, categories)); // 设置子分类return m; // 返回处理后的分类}).collect(Collectors.toList()); // 收集结果到新的列表中return collect; // 返回构建好的树形结构列表}/*** 获取子节点列表* @param tree 父节点* @param list 分类列表* @return 子节点列表*/public static List<DeptEntity> getChildrenList(DeptEntity tree, List<DeptEntity> list) {// 从列表中过滤出所有父分类ID等于给定父分类ID的分类List<DeptEntity> children = list.stream().filter(item -> Objects.equals(item.getParentId(), tree.getDeptId())).map((item) -> { // 对每个子分类进行处理item.setChildren(getChildrenList(item, list)); // 递归设置子分类的子分类return item; // 返回处理后的子分类}).collect(Collectors.toList()); // 收集结果到新的列表中return children; // 返回构建好的子分类列表}
}