如何创建具有相同元素n次的列表?

时间:2012-09-06 12:29:32

标签: scala

如何创建具有相同元素n次的列表?

手动实施:

scala> def times(n: Int, s: String) =
 | (for(i <- 1 to n) yield s).toList
times: (n: Int, s: String)List[String]

scala> times(3, "foo")
res4: List[String] = List(foo, foo, foo)

还有内置的方法吗?

4 个答案:

答案 0 :(得分:139)

请参阅scala.collection.generic.SeqFactory.fill(n:Int)(elem: =>A)收集数据结构,例如SeqStreamIterator等,扩展:

scala> List.fill(3)("foo")
res1: List[String] = List(foo, foo, foo)

警告它在Scala 2.7中不可用。

答案 1 :(得分:8)

像这样使用tabulate

List.tabulate(3)(_ => "foo")

答案 2 :(得分:6)

(1 to n).map( _ => "foo" )

像魅力一样。

答案 3 :(得分:1)

我有另一个模仿flatMap的答案我认为(发现这个解决方案在应用duplicateN时返回Unit)

 implicit class ListGeneric[A](l: List[A]) {
  def nDuplicate(x: Int): List[A] = {
    def duplicateN(x: Int, tail: List[A]): List[A] = {
      l match {
       case Nil => Nil
       case n :: xs => concatN(x, n) ::: duplicateN(x, xs)
    }
    def concatN(times: Int, elem: A): List[A] = List.fill(times)(elem)
  }
  duplicateN(x, l)
}

}

def times(n: Int, ls: List[String]) = ls.flatMap{ List.fill(n)(_) }

但这适用于预定的List,并且您希望每个元素重复n次