最佳实施方式" zipLongest"在斯卡拉

时间:2014-08-21 13:40:35

标签: scala collections functional-programming

我需要在Scala中实现"zipLongest"函数;也就是说,将两个序列组合在一起,如果一个比另一个长,则使用默认值。 (与标准zip方法不同,后者将截断为最短的序列。)

我直接按照以下方式实施了它:

def zipLongest[T](xs: Seq[T], ys: Seq[T], default: T): Seq[(T, T)] = (xs, ys) match {
  case (Seq(), Seq())           => Seq()
  case (Seq(), y +: rest)       => (default, y) +: zipLongest(Seq(), rest, default)
  case (x +: rest, Seq())       => (x, default) +: zipLongest(rest, Seq(), default)
  case (x +: restX, y +: restY) => (x, y) +: zipLongest(restX, restY, default)
}

有更好的方法吗?

2 个答案:

答案 0 :(得分:8)

使用zipAll

scala> val l1 = List(1,2,3)
l1: List[Int] = List(1, 2, 3)

scala> val l2 = List("a","b")
l2: List[String] = List(a, b)

scala> l1.zipAll(l2,0,".")
res0: List[(Int, String)] = List((1,a), (2,b), (3,.))

如果要对第一个和第二个seq使用相同的默认值:

scala> def zipLongest[T](xs:Seq[T], ys:Seq[T], default:T) = xs.zipAll(ys, default, default)
zipLongest: [T](xs: Seq[T], ys: Seq[T], default: T)Seq[(T, T)]

scala> val l3 = List(4,5,6,7)
l3: List[Int] = List(4, 5, 6, 7)

scala> zipLongest(l1,l3,0)
res1: Seq[(Int, Int)] = List((1,4), (2,5), (3,6), (0,7))

答案 1 :(得分:6)

你可以做一个oneliner:

xs.padTo(ys.length, x).zip(ys.padTo(xs.length, y))