list - 如何实现列表的 'takeUntil'？

我想找到所有等于第一个7的项目:

val list = List(1,4,5,2,3,5,5,7,8,9,2,7,4)

我的解决方案是:

list.takeWhile(_ != 7) ::: List(7)

结果是:

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

还有其他解决方案吗？

最佳答案

一线耐心:

List(1, 2, 3, 7, 8, 9, 2, 7, 4).span(_ != 7) match {case (h, t) => h ::: t.take(1)}

更通用的版本:

它以任何谓词为参数。使用span来完成主要工作:

  implicit class TakeUntilListWrapper[T](list: List[T]) {
    def takeUntil(predicate: T => Boolean):List[T] = {
      list.span(predicate) match {
        case (head, tail) => head ::: tail.take(1)
      }
    }
  }

  println(List(1,2,3,4,5,6,7,8,9).takeUntil(_ != 7))
  //List(1, 2, 3, 4, 5, 6, 7)

  println(List(1,2,3,4,5,6,7,8,7,9).takeUntil(_ != 7))
  //List(1, 2, 3, 4, 5, 6, 7)

  println(List(1,2,3,4,5,6,7,7,7,8,9).takeUntil(_ != 7))
  //List(1, 2, 3, 4, 5, 6, 7)

  println(List(1,2,3,4,5,6,8,9).takeUntil(_ != 7))
  //List(1, 2, 3, 4, 5, 6, 8, 9)

尾递归版本。

仅为了说明替代方法，它没有比以前的解决方案更有效。

implicit class TakeUntilListWrapper[T](list: List[T]) {
  def takeUntil(predicate: T => Boolean): List[T] = {
    def rec(tail:List[T], accum:List[T]):List[T] = tail match {
      case Nil => accum.reverse
      case h :: t => rec(if (predicate(h)) t else Nil, h :: accum)
    }
    rec(list, Nil)
  }
}

关于list - 如何实现列表的 'takeUntil'？，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/33602714/