如何在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。