Scala中扁平化嵌套Complex对象的实现方法咨询

嘿，这个问题的核心其实是两个关键点：一是把所有嵌套层级的Complex节点全部提取出来，二是每个提取出来的节点的nested字段必须清空为空列表。之前你尝试的简单扁平化方案可能只处理了列表的层级展开，没考虑修改每个节点的nested属性，所以达不到预期效果。

实现思路 #

我们需要遍历每个Complex节点：

1. 对当前节点创建一个副本，将其nested字段替换为空列表List.empty（或Nil）；
2. 递归处理当前节点的nested列表中的所有子节点，重复步骤1；
3. 将当前节点的副本与所有子节点的扁平化结果合并，最终得到全量的扁平列表。

递归实现方案 #

这个版本代码简洁直观，适合大多数场景：

case class Complex(id: Long, name: String, nested: Seq[Complex])

def flattenComplexList(complexList: List[Complex]): List[Complex] = {
  complexList.flatMap { complex =>
    // 生成当前节点的副本，清空nested
    val currentNode = complex.copy(nested = Nil)
    // 合并当前节点和子节点的扁平化结果
    currentNode :: flattenComplexList(complex.nested.toList)
  }
}

测试示例 #

用你给出的嵌套列表测试：

val stuff = List(
  Complex(1, "name1", List(
    Complex(2, "name2", List()),
    Complex(3, "name3", List(
      Complex(4, "name4", List())
    ))
  ))
)

val flattened = flattenComplexList(stuff)
// 输出结果：List(Complex(1,"name1",List()), Complex(2,"name2",List()), Complex(3,"name3",List()), Complex(4,"name4",List()))

尾递归优化版本（适合深层嵌套） #

如果你的嵌套层级非常深，递归可能导致栈溢出，这时可以用尾递归版本优化：

import scala.annotation.tailrec

@tailrec
def flattenComplexTailRec(remaining: List[Complex], acc: List[Complex] = Nil): List[Complex] = remaining match {
  case Nil => acc.reverse // 反转累加器保证顺序正确
  case head :: tail =>
    // 清空当前节点的nested并加入累加器
    val updatedAcc = head.copy(nested = Nil) :: acc
    // 将当前节点的子节点放到待处理列表头部，继续遍历
    flattenComplexTailRec(head.nested.toList ++ tail, updatedAcc)
}

调用方式和结果和递归版本一致：

val flattenedTailRec = flattenComplexTailRec(stuff)
// 输出和之前完全相同的结果

为什么之前的方案不行？ #

那些通用的列表扁平化方法（比如flatten）只是把嵌套的列表结构展开，但不会修改每个Complex对象的nested字段——我们需要的不仅是把所有节点拉平，还要每个节点的nested都变成空列表，所以必须针对Complex类的结构做定制化处理。

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