Scala 按有限的值序列对集合进行分组

Question

对于

val xs = (1 to 9).toArray

我们可以像这样每两个连续的项目分组，

xs.grouped(2)

然而，给定一个有限的值序列，例如

val gr = Seq(3,2,1)

如何根据 gr 对 xs 进行分组，以便

xs.grouped(gr)
res: Array(Array(1,2,3), Array(4,5), Array(6), Array(7,8,9))

Answer 1

以下函数生成您要查找的结果，但我怀疑可能有更好的方法：

def grouped[T](what: Seq[T], by: Seq[Int]) = {

  def go(left: Seq[T], nextBy: Int, acc: List[Seq[T]]): List[Seq[T]] = (left.length, by(nextBy % by.length)) match {
    case (n, sz) if n <= sz => left :: acc
    case (n, sz) => go(left.drop(sz), nextBy+1, left.take(sz) :: acc)
  }

  go(what, 0, Nil).reverse
}

Answer 2

请考虑以下解决方案：

def groupBySeq[T](arr:Array[T], gr:Seq[Int]) = {
  val r = gr.foldLeft((arr, List[Array[T]]())) {
    case ((rest, acc), item) => (rest.drop(item), rest.take(item)::acc)
  }
  (r._1::r._2).reverse
}

Scala 按有限的值序列对集合进行分组

Scala grouping a collection by a finite sequence of values

scala

grouping

scala-collections