如何在Scala中定义递归列表类型？

Question

我想创建一个flatten函数，它将获取各种深度的List并将其转换为平面列表。

例如，对于整数，它可以采取List(1, List(2, 3))并返回List(1, 2, 3)。

如何正确声明此函数？

def flatten(list: List[???]): List[T]

Answer 1

看起来你必须使用Any因为列表的深度是未知的。

def flatten(input: List[Any]): List[Any] = 
    input match { 
     case Nil => Nil 
     case head :: tail => 
     head match { 
      case list: List[_] => flatten(list) ::: flatten(tail) 
      case elem => elem :: flatten(tail) 
     } 
    } 

scala> flatten(List(1, List(2, 3))) 
res0: List[Any] = List(1, 2, 3) 

如果你想看到几个实施选项检查here和测试here。