如何合并 2 个列表,以使结果列表在 Scala 中以交替方式包含 2 个列表的元素。
输入:
val list1 = List("Mary", "a", "lamb")
val list2 = List("had", "little")
输出:
List("Mary", "had", "a", "little", "lamb")
您正在寻找的通常称为“intersperse”或“intercalate”,有几种方法可以做到:
def intersperse[A](a : List[A], b : List[A]): List[A] = a match {
case first :: rest => first :: intersperse(b, rest)
case _ => b
}
你也可以使用scalaz
import scalaz._
import Scalaz._
val lst1 = ...
val lst2 = ...
lst1 intercalate lst2
编辑:您还可以执行以下操作:
lst1.zipAll(lst2,"","") flatMap { case (a, b) => Seq(a, b) }
想一想,我相信最后一个解决方案是我最喜欢的,因为它最简洁但仍然清晰。如果您已经在使用 Scalaz,我会使用第二种解决方案。然而,第一个也是非常可读的。
为了使这个答案更完整,添加@Travis Brown 的通用解决方案:
list1.map(List(_)).zipAll(list2.map(List(_)), Nil, Nil).flatMap(Function.tupled(_ ::: _))
val list1 = List("Mary", "a", "lamb")
val list2 = List("had", "little")
def merge1(list1: List[String], list2: List[String]): List[String] = {
if (list1.isEmpty) list2
else list1.head :: merge(list2, list1.tail)
}
def merge2(list1: List[String], list2: List[String]): List[String] = list1 match {
case List() => list2
case head :: tail => head :: merge(list2, tail)
}
merge1(list1, list2)
merge2(list1, list2)
//> List[String] = List(Mary, had, a, little, lamb)
list1.zipAll(list2,"","").flatMap(_.productIterator.toList).filter(_ != "")
你可以这样做:
def alternate[A]( a: List[A], b: List[A] ): List[A] = {
def first( a: List[A], b: List[A] ): List[A] = a match {
case Nil => Nil
case x :: xs => x :: second( xs, b )
}
def second( a: List[A], b: List[A] ): List[A] = b match {
case Nil => Nil
case y :: ys => y :: first( a, ys )
}
first( a, b )
}