笛卡尔积(组合)与空Seq

时间:2017-08-10 20:33:41

标签: scala scala-collections

我想要scala中三个Seqs的笛卡尔积(组合)。其中任何一个都可以是空的。我知道任何Set with the empty set的笛卡尔积都是空集。我想要一个提供默认值的工作(在下面的例子中为“”)。这是骨架代码。

val letters = Seq("a", "b")
val numbers = Seq("1")
val empty: Seq[String] = Nil
// Desired output = Seq(("a", "1", ""), ("b", "1", ""))

// Cartesian product
// What I'm trying which returns List()
for {
 l <- letters
 n <- numbers
 e <- empty
} yield {
    (l, n, e)
}

到目前为止,我已尝试过lift和orElse(Seq(“”))。 orElse的问题在于它返回了一个PartialFunction,我失去了map或flatMap的能力。

1 个答案:

答案 0 :(得分:2)

如果您事先知道默认值,则可以使用默认值创建单个元素的列表:

def nonEmptyOrDefault[A](xs: Seq[A], default: A): Seq[A] =
  if (xs.isEmpty) Seq(default)
  else xs

for {
  l <- nonEmptyOrDefault(letters, "")
  n <- nonEmptyOrDefault(numbers, "")
  e <- nonEmptyOrDefault(empty, "")
} yield (l, n, e)

或者您可以使用确保非空虚的集合

class NESeq[+A] private (private val xs: Iterable[A]) {
  // TODO implement flatMap, map, ++, etc.
}

object NESeq {
  def fromIterable[A](xs: Iterable[A]): Option[NESeq[A]] =
    if (xs.isEmpty) None
    else Some(new NESeq(xs))
}