for循环中相同集合的迭代器

Question

假设我想生成集合子集的所有组合。由于subset 返回iterator，我不想将其转换为严格的。

def gen(A: Set[Int]) = {
  val it0 = A.subsets
  val it1 = A.subsets
  for(a <- it0; b <- it1) yield (a,b)
}

但这不是我想要的。例如gen(Set(1,2,3)).foreach(println) 返回：

(Set(),Set())
(Set(),Set(1))
(Set(),Set(2))
(Set(),Set(3))
(Set(),Set(1, 2))
(Set(),Set(1, 3))
(Set(),Set(2, 3))
(Set(),Set(1, 2, 3))

似乎只有第二个迭代器迭代所有子集。为什么会有这样的行为，有没有一种很好的方法可以避免这种情况？

Answer 1

请注意，it0 和 it1 是 Iterators。你不能像这样使用迭代器：

val it0 = Iterator(1, 2)
val it1 = Iterator(1, 2)

(for { a <- it0; b <- it1 } yield (a, b)).toList
// List[(Int, Int)] = List((1,1), (1,2))

这里的原因是你不能重复Iterator，Iterator 是可变的。对于it0 的第一个元素，您已经迭代了it1，所以it1 对于it0 的下一个元素是空的。

您应该为第一个迭代器的每个元素重新创建第二个迭代器：

def gen(A: Set[Int]) = 
  for{
    a <- A.subsets
    b <- A.subsets
  } yield (a,b)

或将Iterator 转换为不可变集合：

def gen(A: Set[Int]) = {
  val it = A.subsets.toSeq
  for(a <- it; b <- it) yield (a,b)
}