如何基于Id层级将数据库返回的扁平化ObjectA列表重构为层级结构？

嘿，这个场景太常见了——存储的时候扁平化，查出来要还原成嵌套结构，其实用ID的层级规则就能轻松搞定，我给你拆解下具体思路和步骤：

先明确你的需求背景 #

你提到的对象结构是这样的：

- List
  - ObjectA(Id=01)
    - SubList
      - ObjectA(Id=01-01)
      - ObjectA(Id=01-02)
        - SubList
          - ObjectA(Id=01-02-01)
  - ObjectA(Id=02)
    - SubList
      - ObjectA(Id=02-01)

现在从数据库拿到的是所有ObjectA的扁平化列表，每个行对应一个对象，需要把它们按ID的层级关系重新组装成上面的嵌套结构。

核心解决方案思路 #

关键是先建立一个ID到对象的快速映射，然后通过解析ID找到每个对象的父节点，把它挂到父节点的SubList里，最后收集所有顶级节点（没有父节点的）就是根List了。这个方法时间复杂度是O(n)，效率很高。

具体步骤（用伪代码+示例说明） #

假设你的扁平化数据是一个数组flatObjects，每个ObjectA包含Id属性和空的SubList（需要我们填充）。

1. 第一步：创建ID映射字典
   先把所有对象放进一个字典里，键是对象的ID，值是对象本身。这样后续找父对象的时候能直接定位，不用每次遍历整个列表：

   # 以Python为例，其他语言思路一致：Java用HashMap，JS用Map，C#用Dictionary
   id_map = {obj.Id: obj for obj in flatObjects}
   

2. 第二步：遍历对象，挂载到父节点
   对每个对象，拆分它的ID来判断是顶级节点还是子节点：

   • 如果ID没有-（比如01、02），说明是顶级节点，直接加入根列表。
   • 如果ID有-，把最后一段去掉就是父ID（比如01-02-01去掉最后一段01，得到父ID01-02），然后从映射字典里取出父对象，把当前对象添加到父对象的SubList中。

   root_list = []
   for obj in flatObjects:
       id_parts = obj.Id.split('-')
       if len(id_parts) == 1:
           # 顶级对象，加入根列表
           root_list.append(obj)
       else:
           # 生成父ID：去掉ID的最后一部分
           parent_id = '-'.join(id_parts[:-1])
           parent_obj = id_map.get(parent_id)
           # 这里可以加个判断，防止父ID不存在的情况
           if parent_obj:
               parent_obj.SubList.append(obj)
           else:
               # 可选：记录日志或者处理异常，比如把这个对象作为顶级节点
               print(f"Warning: Parent ID {parent_id} not found for object {obj.Id}")
               root_list.append(obj)
   

3. 第三步：验证结果
   跑完上面的代码后，root_list就是你需要的嵌套层级结构了，每个对象的SubList都已经正确填充了子对象。

额外注意点 #

• 一定要确保ID的格式严格符合规则，比如子对象的父ID一定存在于列表中，如果有异常ID（比如父ID找不到），记得加异常处理，避免程序崩溃。
• 如果你的ObjectA的SubList初始不是空列表，记得先清空再填充，防止重复数据。
• 这个思路完全通用，不管你用什么编程语言，只要能实现“字典映射”和“字符串拆分”这两个操作就行。

内容的提问来源于stack exchange，提问作者Madan Thunderfist