Zip два списка разной длины с элементом по умолчанию для заполнения

Предположим, у нас есть следующие списки разного размера:

val list1 = ("a", "b", "c")
val list2 = ("x", "y")

Теперь я хочу объединить эти 2 списка и создать новый список со строкой элементов, объединяемых:

val desiredResult = ("ax", "by", "c")

пробовал

val wrongResult = (list1, list2).zipped map (_ + _)

как предложил здесь, но это не работает по назначению, потому что zip отбрасывает те элементы более длинного списка, которые не могут быть сопоставлены.

как я могу решить эту проблему? Есть ли способ zip списки и дать "элемент по умолчанию" (например, пустая строка в этом случае), если один список длиннее?

11
functional-programming scala

2 ответов

метод, который вы ищете является .zipAll:

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

scala> val list2 = List("x", "y")
list2: List[String] = List(x, y)

scala> list1.zipAll(list2, "", "")
res0: List[(String, String)] = List((a,x), (b,y), (c,""))

.zipAll принимает 3 аргумента:

  • iterable для zip с
  • значение по умолчанию if this (сборник .zipAll вызывается on) короче
  • значение по умолчанию, если другая коллекция короче
29

на основе API zipAll - это путь, но вы можете ее реализовать (как упражнение), например, следующим образом:

implicit class OpsSeq[A,B](val xs: Seq[A]) extends AnyVal {
  def zipAll2(ys: Seq[B], xDefault: A, yDefault: B) = {
    val xs2 = xs ++ Seq.fill(ys.size-xs.size)(xDefault)
    val ys2 = ys ++ Seq.fill(xs.size-ys.size)(yDefault)
    xs2.zip(ys2) 
  }
}

отсюда, например,

Seq(1,2).zipAll2(Seq(3,4,5),10,20)
List((1,3), (2,4), (10,5))

и

list1.zipAll2(list2, "", "")
List((a,x), (b,y), (c,""))

рекурсивная версия,

def zipAll3[A,B](xs: Seq[A], ys: Seq[B], xd: A, yd: B): Seq[(A,B)] = {
  (xs,ys) match {
    case (Seq(),    Seq())    => Seq()
    case (x +: xss, Seq())    => (x,yd) +: zipAll3(xss, Seq(), xd, yd)
    case (Seq(),    y +: yss) => (xd,y) +: zipAll3(Seq(), yss, xd, yd)
    case (x +: xss, y +: yss) => (x,y) +: zipAll3(xss, yss, xd, yd) 
  }
}

по умолчанию xd по умолчанию yd значения.

3