Question

我需要在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)
}

有更好的方法吗？

Answer 1

使用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))

Answer 2

你可以做一个oneliner：

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

最佳实施方式＆＃34; zipLongest＆＃34;在斯卡拉

2 个答案: