Scala spłaszczyć Listę

Question

Chcę napisać funkcję, która łączy listę.

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, zwraca List(1, 1, 2, List(3, List(5, 8))) ale powinno być List(1, 1, 2, 3, 5, 8).

Czy możesz dać mi wskazówkę?

Answer 1

Przez kasowania linii 4

case head :: Nil => List(head) 

Dostaniesz właściwą odpowiedź.

Pomyśl o przypadku testowego

List(List(List(1))) 

z linią 4 ostatni element na liście nie będą przetwarzane

Answer 2

Nie trzeba do gniazda swoje wypowiedzi meczu. Zamiast tego dopasuj na miejscu tak:

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) 
    } 

Answer 3

Mój, odpowiednik rozwiązania SDJMcHattie's.

def flatten(xs: List[Any]): List[Any] = xs match { 
    case List() => List() 
    case (y :: ys) :: yss => flatten(y :: ys) ::: flatten(yss) 
    case y :: ys => y :: flatten(ys) 
    } 

Answer 4

def flatten(ls: List[Any]): List[Any] = ls flatMap { 
    case ms: List[_] => flatten(ms) 
    case e => List(e) 
    }