交换列表中 x 元素的正确方法

Question

我开始学习Scala语言，我有一个问题。您认为以函数式风格交换 List 中第一个和最后一个 x 元素是否是正确的方法？

def swap(l: List[Any], x: Int) = {
  val l1 = l.take(x)
  val l2 = l.slice(x, l.length - x)
  val l3 = l.takeRight(x)
  l3 ::: l2 ::: l1
}

如果 x 超过列表长度的一半，发生什么并不重要。我有兴趣了解算法。

Answer 1

这段代码是正确的并且具有合理的功能风格。它不是最有效的，因为它必须遍历列表四次才能创建 l1 到 l3 片段。另外，您可能想保留列表包含的类型，因此略有改进是：

def swap[A](l: List[A], x: Int) = {
  val (l1,rest) = l.splitAt(x)
  val (l2,l3) = rest.splitAt(rest.length-x)
  l3 ::: l2 ::: l1
}

Answer 2

我尝试了一下，效果很好：

scala> swap(List(1, 2, 3, 4, 5),2)
res0: List[Any] = List(4, 5, 3, 1, 2)

你自己提供的代码有什么问题吗？