scala - 如何从列表中删除 2 个或多个重复项并保持其初始顺序？

Question

假设我们有一个 Scala 列表:

val l1 = List(1, 2, 3, 1, 1, 3, 2, 5, 1)

我们可以使用以下代码轻松删除重复项:

l1.distinct

或

l1.toSet.toList

但是如果我们只想删除超过 2 个重复项怎么办？因此，如果有超过 2 个具有相同值的元素，我们只保留两个并删除其余的。

我可以通过以下代码实现它:

l1.groupBy(identity).mapValues(_.take(2)).values.toList.flatten

这给了我结果:

List(2, 2, 5, 1, 1, 3, 3)

元素被删除，但剩余元素的顺序与这些元素在初始列表中的显示顺序不同。如何进行此操作并保持原始列表的顺序？

所以 l1 的结果应该是:

List(1, 2, 3, 1, 3, 2, 5)

Answer 1

不是最有效的。

scala> val l1 = List(1, 2, 3, 1, 1, 3, 2, 5, 1)
l1: List[Int] = List(1, 2, 3, 1, 1, 3, 2, 5, 1)

scala> l1.zipWithIndex.groupBy( _._1 ).map(_._2.take(2)).flatten.toList.sortBy(_._2).unzip._1
res10: List[Int] = List(1, 2, 3, 1, 3, 2, 5)