Scala flatten List
Chcę napisać funkcję, która pasuje do listy.
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)))))
}
}
Nie wiem, dlaczego to nie działa, wracaList(1, 1, 2, List(3, List(5, 8)))
ale tak powinno byćList(1, 1, 2, 3, 5, 8)
.
Czy możesz dać mi wskazówkę?