实现这个简单的类拆分函数的惯用 scala 方式

Question

我想实现一个具有以下特点的函数：

• 将一个array (Array[A]) 和多个pieces 作为参数返回(Int)
• 返回分割成pieces 块的array 数组。
• 如果可能，所有片段的长度应该相等，否则第一个片段应该比最后一个片段长一个元素（如果array 的长度不是pieces 的倍数）

在 Haskell 中，我会按照这些思路编写一些代码：

split :: [a] -> Int -> [[a]]
split list pieces = go list (length list `div` pieces)
 where
  go xs n | l > n && m == 0 = take n xs : go (drop n xs) n
          | l > n           = take (n + 1) xs : go (drop (n + 1) xs) n
          | otherwise       = [xs]
   where
    l = length xs
    m = l `mod` n

虽然在 Scala 中，我在编写这个（基本）函数时遇到了很多困难。对于递归，Array 似乎不太适应。然后，if 结构允许我实现我在 haskell 中使用的那种守卫，不允许代替表达式，这对我来说似乎很奇怪。我遇到的另一个问题是我不知道如何使我的 scala 代码多态（就像我的 Haskell 代码一样）。最后但并非最不重要的一点是，我不明白如何不仅制作 a 等价多态，还制作 Array 实例，因为 Scala 中有许多基本集合。

我会接受完整的解释解决方案或简单的提示来回答我的误解。

Answer 1

翻译非常简单。我没有分别调用take 和drop，而是使用splitAt，这可能更有效。当然，你可以使这个尾递归。而且我还认为，“惯用的”Scala 可能在方法中使用IndexedSeq.newBuilder，这可能再次更有效，但我想这不是你想要的。

请注意，Scala 的数组是可变的，因此您需要Vector 或更通用的IndexedSeq。

import collection.immutable.{IndexedSeq => ISeq}

def split[A](seq: ISeq[A], pieces: Int): ISeq[ISeq[A]] = {
  val n = seq.size / pieces
  def loop(xs: ISeq[A]): ISeq[ISeq[A]] = {   
    val l = xs.size
    if(l > n) {
      val m = l % n
      val (begin, end) = xs.splitAt(if(m == 0) n else n + 1)
      begin +: loop(end)
    } else ISeq(xs)
  }
  loop(seq)
}

split(1 to 20, 3) // correct