Question

从Scala中的列表中删除第一个对象的最佳方法是什么？

来自Java，我习惯于使用List.remove(Object o)方法从列表中删除第一次出现的元素。既然我在Scala中工作，我希望该方法返回一个新的不可变List而不是改变一个给定的列表。我可能还希望remove()方法采用谓词而不是对象。总之，我希望找到这样的方法：

/**
 * Removes the first element of the given list that matches the given
 * predicate, if any.  To remove a specific object <code>x</code> from
 * the list, use <code>(_ == x)</code> as the predicate.
 *
 * @param toRemove
 *          a predicate indicating which element to remove
 * @return a new list with the selected object removed, or the same
 *         list if no objects satisfy the given predicate
 */
def removeFirst(toRemove: E => Boolean): List[E]

当然，我可以通过几种不同的方式实现这种方法，但是没有一种方法能够突然出现，因为它显然是最好的。我宁愿不将我的列表转换为Java列表（甚至转换为Scala可变列表）并再次返回，尽管这肯定会有效。我可以使用List.indexWhere(p: (A) ⇒ Boolean)：

def removeFirst[E](list: List[E], toRemove: (E) => Boolean): List[E] = {
  val i = list.indexWhere(toRemove)
  if (i == -1)
    list
  else
    list.slice(0, i) ++ list.slice(i+1, list.size)
}

但是，使用带有链表的索引通常不是最有效的方法。

我可以写一个更有效的方法：

def removeFirst[T](list: List[T], toRemove: (T) => Boolean): List[T] = {
  def search(toProcess: List[T], processed: List[T]): List[T] =
    toProcess match {
      case Nil => list
      case head :: tail =>
        if (toRemove(head))
          processed.reverse ++ tail
        else
          search(tail, head :: processed)
    }
  search(list, Nil)
}

尽管如此，这并不完全简洁。似乎很奇怪，现有的方法不会让我有效而简洁地完成这项工作。所以，我错过了什么，或者我的最后一个解决方案真的很好吗？

Answer 1

您可以使用span稍微清理代码。

scala> def removeFirst[T](list: List[T])(pred: (T) => Boolean): List[T] = {
     |   val (before, atAndAfter) = list span (x => !pred(x))
     |   before ::: atAndAfter.drop(1)
     | }
removeFirst: [T](list: List[T])(pred: T => Boolean)List[T]

scala> removeFirst(List(1, 2, 3, 4, 3, 4)) { _ == 3 }
res1: List[Int] = List(1, 2, 4, 3, 4)

Scala Collections API overview是了解一些鲜为人知的方法的好地方。

Answer 2

这种情况有点可变性很长一段时间：

def withoutFirst[A](xs: List[A])(p: A => Boolean) = {
  var found = false
  xs.filter(x => found || !p(x) || { found=true; false })
}

这很容易推广到删除与谓词匹配的第一个n项。（i<1 || { i = i-1; false }）

您也可以自己编写过滤器，但此时使用span几乎肯定会更好，因为如果列表很长，此版本将溢出堆栈：

def withoutFirst[A](xs: List[A])(p: A => Boolean): List[A] = xs match {
  case x :: rest => if (p(x)) rest else x :: withoutFirst(rest)(p)
  case _ => Nil
}

并且其他任何事情都比span更复杂，没有任何明显的好处。

如何从Scala中的列表中删除第一个对象？

2 个答案: