在递归展平方法实现中出现类型错误

Question

我正在尝试学习 scala。今天我正在尝试编写一个简单的递归方法来展平嵌套列表。我知道有一个可以调用的 flatten 函数，但我正在尝试从头开始。

我收到类型不匹配错误，我正在尝试了解原因。 'A' 是什么类型的变量。

def flatten[A](lst:List[List[A]):List[A] = lst match{

case Nil=> Nil
case (h:List[A])::tail=> flatten(h)::flatten(tail)
case h :: tail=> flatten(tail)
}

Answer 1

flatten(h) 是一个编译器错误，因为h 是一个List[A]，但flatten 需要一个嵌套的List[List[A]]。尝试简单地将h 与扁平化tail 的结果连接起来：

def flatten[A](lst: List[List[A]]): List[A] = lst match {
  case Nil => Nil
  case h :: tail => h ::: flatten(tail)
}

例子：

scala> flatten(List(List("a", "b"), List("c", "d")))
res0: List[String] = List(a, b, c, d)

scala> flatten(List(List(1, 2), List(3, 4)))
res1: List[Int] = List(1, 2, 3, 4)