将List对象列表转为树形结构

Java 中将平铺 List 转换为树形结构

在社交标签、菜单、分类等场景中，数据库通常以平铺的父子关系表存储数据（每条记录带一个 parentId）。前端渲染时往往需要树形结构。本文给出一个基于 Java 的实体定义与“List 转 Tree”实现示例。

一、实体类 SocialTag

标签实体继承基础实体，关键字段包括主键 id、标签名称 tagName、父级 id parentId、标签等级 levelNo、排序号 sortNo、可用状态 enableFlag、删除标识 deleteFlag，并额外维护一个非表字段 childSocialTagList 存放子节点：

1. package com.nijia.vip.social.entity;
   
2. 
   
3. import java.util.ArrayList;
   
4. import java.util.List;
   
5. 
   
6. import com.baomidou.mybatisplus.annotation.IdType;
   
7. import com.baomidou.mybatisplus.annotation.TableField;
   
8. import com.baomidou.mybatisplus.annotation.TableId;
   
9. import com.nijia.base.entity.BaseEntity;
   
10. 
    
11. import io.swagger.annotations.ApiModelProperty;
    
12. import lombok.Data;
    
13. 
    
14. /**
    
15. * 社交--个人标签(SocialTag)表实体类
    
16. *
    
17. * @author whp
    
18. * @since 2022-06-23 14:15:53
    
19. */
    
20. @SuppressWarnings("serial")
    
21. @Data
    
22. public class SocialTag extends BaseEntity {
    
23.     private static final long serialVersionUID = 1L;
    
24. 
    
25.     @TableId(value = "id", type = IdType.ASSIGN_ID)
    
26.     @ApiModelProperty("主键")
    
27.     private Long id;
    
28. 
    
29.     @ApiModelProperty("标签名称")
    
30.     private String tagName;
    
31. 
    
32.     @ApiModelProperty("父级id")
    
33.     private Long parentId;
    
34. 
    
35.     @ApiModelProperty("标签等级【1-1级，2-2级】")
    
36.     private Integer levelNo;
    
37. 
    
38.     @ApiModelProperty("排序号")
    
39.     private Integer sortNo;
    
40. 
    
41.     @ApiModelProperty("可用状态【1-可用，0-不可用】")
    
42.     private Integer enableFlag;
    
43. 
    
44.     @ApiModelProperty("删除标识【1-删除，0-正常】")
    
45.     private Integer deleteFlag;
    
46. 
    
47.     @TableField(exist = false)
    
48.     private List<SocialTag> childSocialTagList = new ArrayList<SocialTag>();
    
49. }

复制代码

二、List 转 Tree 的核心算法

思路是先把 List 转成以 id 为键的 Map，再遍历每个节点：若其 parentId 对应的父节点不存在，说明它是顶级节点，加入结果集；否则把它挂到父节点的 childSocialTagList 中。

1. private static List<SocialTag> convertListToTree(List<SocialTag> socialTagList) {
   
2.     List<SocialTag> resultList = new ArrayList<SocialTag>();
   
3.     if (CollectionUtils.isEmpty(socialTagList)) {
   
4.         return resultList;
   
5.     }
   
6. 
   
7.     // List 转为 Map（id -> 节点）
   
8.     Map<Long, SocialTag> socialTagMap =
   
9.             socialTagList.stream().collect(Collectors.toMap(SocialTag::getId, item -> item));
   
10. 
    
11.     for (SocialTag socialTag : socialTagList) {
    
12.         SocialTag parentSocialTag = socialTagMap.get(socialTag.getParentId());
    
13.         // 该节点为空，说明是顶级节点
    
14.         if (parentSocialTag == null) {
    
15.             resultList.add(socialTag);
    
16.             continue;
    
17.         }
    
18.         // 把自己加到父节点对象里
    
19.         parentSocialTag.getChildSocialTagList().add(socialTag);
    
20.     }
    
21.     return resultList;
    
22. }

复制代码

三、构造测试数据并验证

通过 initData() 准备两级标签（如 Java/Springboot、DB/Mysql 等），调用转换后打印 JSON：

1. public static void main(String[] args) {
   
2.     List<SocialTag> resultList = convertListToTree(initData());
   
3.     System.out.println(JavaJsonConvert.javaToJson(resultList));
   
4. }

复制代码

其中 initData() 创建若干 SocialTag，设置 id、parentId、levelNo 与创建时间后返回列表。运行后即可得到嵌套了 childSocialTagList 的树形结果。

小结

该实现时间复杂度为 O(n)，借助 Map 一次遍历即可完成挂载，比嵌套循环查找父节点更高效，适合标签、菜单等一对多层级数据的后端组装。

qidao123.com企服评测-应用市场 编辑提示