13

リストをフラット化する関数を書きたい。

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

ヒントを教えていただけますか？

4

5 に答える 5

33

一致ステートメントをネストする必要はありません。代わりに、次のように適切な場所でマッチングを行います。

  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)
  }
于 2013-04-23T17:12:21.773 に答える
16

私の、SDJMcHattieのソリューションに相当します。

  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)
  }
于 2014-06-09T20:21:07.513 に答える
11

4行目を削除する

case head :: Nil => List(head)

あなたは正しい答えを得るでしょう。

テストケースについて考える

List(List(List(1)))

4行目では、リストの最後の要素は処理されません

于 2012-10-25T01:27:11.803 に答える
2 
  def flatten(ls: List[Any]): List[Any] = ls flatMap {
    case ms: List[_] => flatten(ms)
    case e => List(e)
  }
于 2015-04-10T10:49:57.993 に答える
0

受け入れられたソリューションのこの行を誰かが理解していない場合、またはパターンにタイプで注釈を付けることができることを知らなかった場合：

case (head: List[_]) :: tail => flatten(head) ++ flatten(tail)

次に、型注釈のない同等のものを見てください。

case (y :: ys) :: tail => flatten3(y :: ys) ::: flatten3(tail)
case Nil :: tail => flatten3(tail)

したがって、いくつかの選択肢をよりよく理解するために：

def flatten2(xs: List[Any]): List[Any] = xs match {
  case x :: xs => x match {
    case y :: ys => flatten2(y :: ys) ::: flatten2(xs)
    case Nil => flatten2(xs)
    case _ => x :: flatten2(xs)
  }
  case x => x
}

def flatten3(xs: List[Any]): List[Any] = xs match {
  case Nil => Nil
  case (y :: ys) :: zs => flatten3(y :: ys) ::: flatten3(zs)
  case Nil :: ys => flatten3(ys)
  case y :: ys => y :: flatten3(ys)
}

val yss = List(List(1,2,3), List(), List(List(1,2,3), List(List(4,5,6))))
flatten2(yss) // res2: List[Any] = List(1, 2, 3, 1, 2, 3, 4, 5, 6) 
flatten3(yss) // res2: List[Any] = List(1, 2, 3, 1, 2, 3, 4, 5, 6)

ちなみに、2番目に投稿された回答は次のことを行いますが、おそらくあなたは望まないでしょう。

val yss = List(List(1,2,3), List(), List(List(1,2,3), List(List(4,5,6))))
flatten(yss) // res1: List[Any] = List(1, 2, 3, List(), 1, 2, 3, 4, 5, 6)
于 2020-05-19T18:49:29.923 に答える