如何实现列表的’takeUntil’？

发布时间：2020-12-16 09:02:40 所属栏目：安全来源：网络整理

导读：我想要找到之前的所有项目,并且等于前7项： val list = List(1,4,5,2,3,7,8,9,4) 我的解决方案是： list.takeWhile(_ != 7) ::: List(7) 结果是： List(1,7) 还有其他解决方案吗？解决方法不耐烦的单线： List(1,4).span(_ != 7) match {case (h,t) = h ::

我想要找到之前的所有项目,并且等于前7项：

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

我的解决方案是：

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

结果是：

List(1,7)

还有其他解决方案吗？

解决方法

不耐烦的单线：

List(1,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,6,9).takeUntil(_ != 7))
  //List(1,7)

  println(List(1,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)
  }
}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!