Scala 扁平化列表

Question

我想编写一个扁平化列表的函数。

object Flat {
  def flatten[T](list: List[T]): List[T] = list match {
    case Nil => Nil
    case head :: Nil => List(head)
    case head :: tail => (head match {
      case l: List[T] => flatten(l)
      case i => List(i)
    }) ::: flatten(tail)
  }
}

object Main {
  def main(args: Array[String]) = {
    println(Flat.flatten(List(List(1, 1), 2, List(3, List(5, 8)))))
  }
}

我不知道为什么它不起作用，它返回 List(1, 1, 2, List(3, List(5, 8)))但应该是 List(1, 1, 2, 3, 5, 8) .

你能给我一个提示吗？

Answer 1

您不需要嵌套匹配语句。而是像这样就地匹配:

  def flatten(xs: List[Any]): List[Any] = xs match {
    case Nil => Nil
    case (head: List[_]) :: tail => flatten(head) ++ flatten(tail)
    case head :: tail => head :: flatten(tail)
  }