将列表拆分为多个具有固定元素数量的列表

Question

如何将元素列表拆分为最多包含 N 个项目的列表？

例如：给定一个包含 7 个元素的列表，创建 4 个组，最后一组可能包含较少的元素。

split(List(1,2,3,4,5,6,"seven"),4)

=> List(List(1,2,3,4), List(5,6,"seven"))

Answer 1

我认为您正在寻找分组。它返回一个迭代器，但您可以将结果转换为列表，

scala> List(1,2,3,4,5,6,"seven").grouped(4).toList
res0: List[List[Any]] = List(List(1, 2, 3, 4), List(5, 6, seven))

Answer 2

使用滑动方法有更简单的方法来完成任务。
它的工作方式是这样的：

val numbers = List(1, 2, 3, 4, 5, 6 ,7)

假设您想将列表分成大小为 3 的较小列表。

numbers.sliding(3, 3).toList

将为您提供

List(List(1, 2, 3), List(4, 5, 6), List(7))

Answer 3

或者，如果您想自己制作：

def split[A](xs: List[A], n: Int): List[List[A]] = {
  if (xs.size <= n) xs :: Nil
  else (xs take n) :: split(xs drop n, n)
}

使用：

scala> split(List(1,2,3,4,5,6,"seven"), 4)
res15: List[List[Any]] = List(List(1, 2, 3, 4), List(5, 6, seven))

编辑：在两年后回顾此内容时，我不会推荐此实现，因为 size 是 O(n)，并且因此此方法的复杂度为 O(n^2)，这可以解释为什么内置方法对于大型列表变得更快，如下面的注释中所述。您可以按如下方式有效实现：

def split[A](xs: List[A], n: Int): List[List[A]] =
  if (xs.isEmpty) Nil 
  else (xs take n) :: split(xs drop n, n)

或者使用 splitAt 甚至（稍微）更有效：

def split[A](xs: List[A], n: Int): List[List[A]] =
  if (xs.isEmpty) Nil 
  else {
    val (ys, zs) = xs.splitAt(n)   
    ys :: split(zs, n)
  }

Answer 4

我添加了 split 方法的尾递归版本，因为有一些关于尾递归与递归的讨论。我使用 tailrec 注释来强制编译器抱怨，以防实现确实不是尾部递归的。我相信尾递归会变成一个循环，因此即使对于一个大列表也不会导致问题，因为堆栈不会无限期地增长。

import scala.annotation.tailrec


object ListSplitter {

  def split[A](xs: List[A], n: Int): List[List[A]] = {
    @tailrec
    def splitInner[A](res: List[List[A]], lst: List[A], n: Int) : List[List[A]] = {
      if(lst.isEmpty) res
      else {
        val headList: List[A] = lst.take(n)
        val tailList : List[A]= lst.drop(n)
        splitInner(headList :: res, tailList, n)
      }
    }

    splitInner(Nil, xs, n).reverse
  }

}

object ListSplitterTest extends App {
  val res = ListSplitter.split(List(1,2,3,4,5,6,7), 2)
  println(res)
}

Answer 5

我认为这是使用 splitAt 而不是 take/drop 的实现

def split [X] (n:Int, xs:List[X]) : List[List[X]] =
    if (xs.size <= n) xs :: Nil
    else   (xs.splitAt(n)._1) :: split(n,xs.splitAt(n)._2)