scala ArrayBuffer 使用谓词删除所有元素

Question

Scala 在过滤不可变序列方面非常优雅：

var l = List(1,2,3,4,5,6)
l = l.filter(_%2==1)

但是如何使用像 ArrayBuffer 这样的可变集合来做到这一点？ 我发现的只是删除单个元素或切片，或从另一个序列中删除元素，但没有删除谓词给出的元素。

编辑： 我希望能找到类似的东西：

trait Removable[A] extends Buffer[A]{ 
  def removeIf(p: A => Boolean){
    var it1 = 0
    var it2 = 0

    while(it2 < length){
      if( p( this(it2) ) ){
        it2 += 1;
      } 
      else {
        this(it1) = this(it2)
        it1 += 1;
        it2 += 1;
      }
    }

    trimEnd(it2-it1)
  }
}

这个过滤器是线性时间的，可以混合到任何缓冲区中，但只有 ArrayBuffer 才有意义，在 ListBuffers 上它会很慢，因为索引确实需要线性时间。

Answer 1

我的猜测是通过构建新缓冲区进行过滤会更有效，因此您通常只需使用 filter 并使用其结果。否则，您可以编写自己的就地过滤方法：

def filterInPlace[A](b: collection.mutable.Buffer[A])(fun: A => Boolean): Unit = {
  var sz = b.size
  var i = 0; while(i < sz) {
    if (fun(b(i))) {
      i += 1
    } else {
      sz -= 1
      b.remove(i)
    }
  }
}

val b = collection.mutable.ArrayBuffer((1 to 6): _ *)
filterInPlace(b)(_ % 2 == 1)
println(b)

Answer 2

关于拥有一组通过执行突变来工作的方法的讨论，但想出一个好的、通用的集非常困难，而且另一方面，对它的需求还不够。

Answer 3

ArrayBuffer 也是如此。所有集合类都有相同的可用方法。

Answer 4

我想出了这个

import scala.collection.mutable

trait BufferUtils {
    import BufferUtils._
    implicit def extendMutableBuffer[T](org: mutable.Buffer[T]): ExtendedBuffer[T] = new ExtendedBuffer(org)
}

object BufferUtils extends BufferUtils {

    implicit class ExtendedBuffer[T](val org: mutable.Buffer[T]) extends AnyVal {
        def removeIf(pred: (T) => Boolean): Unit = {
            // target holds the index we want to move the next element to
            var target = 0

            for (i <- org.indices;
                 elem = org(i)
                 if !pred(elem)) {

                org(target) = elem
                target += 1
            }

            org.remove(target, org.size - target)
        }
    }

}

Answer 5

通常withFilter 已经足够好了，尤其是当 Buffer 最终转换为不可变结构时。没错，它并没有真正删除元素，但至少它没有创建新的 ArrayBuffer 对象。

Answer 6

这对我有用，但仅适用于 clone()，因此仍会创建一个新的 ArrayBuffer :-)

scala> import collection.mutable.ArrayBuffer
import collection.mutable.ArrayBuffer

scala> val buf = ArrayBuffer(1,2,3,4,5,6,7,8,9,10)
buf: scala.collection.mutable.ArrayBuffer[Int] = ArrayBuffer(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

scala> buf.clone foreach { x => if (x > 4) buf -= x }

scala> buf
res1: scala.collection.mutable.ArrayBuffer[Int] = ArrayBuffer(1, 2, 3, 4)

但更好的方法是创建一个仅包含您要删除的元素的新数组（因此不复制整个缓冲区），然后删除它们：

scala> val buf = ArrayBuffer(1,2,3,4,5,6,7,8,9,10)
buf: scala.collection.mutable.ArrayBuffer[Int] = ArrayBuffer(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)

scala> buf filter { _ > 4 } foreach { buf -= _ }

scala> buf
res3: scala.collection.mutable.ArrayBuffer[Int] = ArrayBuffer(1, 2, 3, 4)