如何扩展 Scala 列表以不按显式位置而是按给定谓词/条件启用切片

For

trait Item
case class TypeA(i: Int) extends Item
case class TypeB(i: Int) extends Item

考虑 Scala 项目列表，例如

val myList = List(TypeA(1), TypeB(11), TypeB(12), 
                  TypeA(2), TypeB(21), 
                  TypeA(3), TypeB(31))

目标是定义一个新的slice可以应用到的方法myList并且以谓词或条件作为参数；例如

myList.slice { x => x.isInstanceOf[TypeA] }

会交付

List(List(TypeA(1), TypeB(11), TypeB(12)), 
     List(TypeA(2), TypeB(21)), 
     List(TypeA(3), TypeB(31)))

在此示例中，通过以下方式可以获得相同的结果

myList.slice { case TypeA(x) => x < 10 }

非常感谢。

List已经有一个slice方法 - 它采用开始索引和结束索引之间的元素子集。您正在寻找的是重复应用span method:

def span(p: (A) ⇒ Boolean): (List[A], List[A])

其记录为：

根据谓词将此列表拆分为前缀/后缀对。

注意：c span p 相当于（但可能比）(c takeWhile p, c dropWhile p) 更有效，前提是谓词 p 的计算不会引起任何副作用。

returns:由该列表中元素全部满足 p 的最长前缀和该列表的其余部分组成的对。

您可以通过重复使用此方法和逆谓词来获得所需的内容，并使用额外的逻辑来确保返回的列表都不为空。

import annotation.tailrec

def multiSpan[A](xs: List[A])(splitOn: (A) => Boolean): List[List[A]] = {
  @tailrec
  def loop(xs: List[A], acc: List[List[A]]) : List[List[A]] = xs match {
    case Nil => acc

    case x :: Nil => List(x) :: acc

    case h :: t =>
      val (pre,post) = t.span(!splitOn(_))
      loop(post, (h :: pre) :: acc)
  }
  loop(xs, Nil).reverse
}

UPDATE

根据原始帖子评论中的要求，这里有一个丰富列表的版本，而不是一个独立的方法：

implicit class AddMultispanToList[A](val list: List[A]) extends AnyVal {
  def multiSpan(splitOn: (A) => Boolean): List[List[A]] = {
    @tailrec
    def loop(xs: List[A], acc: List[List[A]]) : List[List[A]] = xs match {
      case Nil => acc

      case x :: Nil => List(x) :: acc

      case h :: t =>
        val (pre,post) = t.span(!splitOn(_))
        loop(post, (h :: pre) :: acc)
    }
    loop(list, Nil).reverse
  }
}

Use as:

myList.multiSpan(_.isInstanceOf[TypeA])