对列表中的相邻元素进行分组

时间:2013-05-30 22:32:18

标签: algorithm scala functional-programming

假设我想编写一个执行此操作的函数:

输入:[1,1,3,3,4,2,2,5,6,6] 输出:[[1,1],[3,3],[4],[2,2],[5],[6,6]]

它将相邻元素分组相同。

这种方法的名称应该是什么?这个操作有标准名称吗?

2 个答案:

答案 0 :(得分:5)

[1,1,3,3,4,2,2,5,6,6]中,类似[1,1]的内容通常被称为运行(如在游程编码中,请参阅RLE in Scala)。因此,我将调用方法groupRuns

答案 1 :(得分:4)

@tailrec
def groupRuns[A](c: Seq[A], acc: Seq[Seq[A]] = Seq.empty): Seq[Seq[A]] = {
  c match {
    case Seq() => acc
    case xs => 
      val (same, rest) = xs.span { _ == xs.head }
      groupRuns(rest, acc :+ same)
  }
}

scala> groupRuns(Vector(1, 1, 3, 3, 4, 2, 2, 5, 6, 6))
res7: Seq[Seq[Int]] = List(Vector(1, 1), Vector(3, 3), Vector(4), Vector(2, 2), Vector(5), Vector(6, 6))