scala - 如何从 ListBuffer 中删除多个索引(快速)？

Question

Scala 中有没有一种好方法可以从 ListBuffer 中删除多个索引(快速)？

示例:

val indicesToDelete = List(4, 1)
val buffer = ListBuffer(a, b, c, d, e)

结果:

ListBuffer(b, c, e)

我找不到可以完成这项工作的预定义函数。

可以对索引进行排序并删除以最高索引开头的元素等，因此不会有任何复杂化。但排序需要O(n * log n)。有没有更快的方法(也许是我错过的预定义的方法)？

更新 1: 应删除现有 ListBuffer 对象中的元素，不应创建新的 ListBuffer 对象。

Answer 1

您必须使用 zipWithIndex，就像其他帖子已经做的那样，因为否则索引会发生变化，您可能会意外删除错误的项目。但我会使用 collect，而不是 foldLeft 或 filter + map，在这种情况下，其作用与 filter + map，但只需一步即可。

buffer.zipWithIndex.collect { case (x,i) if !indicesToDelete.contains(i) => x }

这也可以写成

for {
  (x,i) <- buffer.zipWithIndex
  if !indicesToDelete.contains(i)
} yield x