声明类型不同时的不同行为（Set 与 TreeSet）

Question

    var set = TreeSet(5,4,3,2,1)
    println(set)

    val diffSet: TreeSet[Int] = set
    // if I change above code to val diffSet: Set[Int] = set
    // the result is unsorted set.

    for (i <- diffSet; x = i) {
        println(i)
    }
    println("-" * 20)
    // the above code translates to below and print the same result
    val temp = diffSet.map(i => (i, i))
    for ((i, x) <- temp) {
        println(i)
    }

我的问题是，如果我定义了这样的方法：

def genSet:Set[Int] = {
  TreeSet(5, 4, 3, 2, 1)
}

当我想使用 for 循环时，

for (i <- genSet; x = i + 1) {
  println(x)
}

结果是未排序的，如何在不更改 genSet 的返回类型的情况下修复此行为。如果我像下面这样使用for循环，那就没问题了，但我希望保留上面的代码风格。

for (i <- genSet) {
  val x = i + 1
  println(x)
}

Answer 1

为什么 map 版本最终未排序

map 方法（使用我们将调用 func 的函数调用）采用隐式 CanBuildFrom 参数，除了考虑 func 返回的类型以选择适当的返回类型之外，还考虑调用 map 的集合类型。这用于使 Map.map[Int] 或 BitSet.map[String] 做正确的事情（返回通用列表），而 Map.map[ (String,Int)] 或 BitSet.map[Int] 也做正确的事情（返回一个 Map 和一个 BitSet ） 分别。

CanBuildFrom 是在编译时选择的，因此必须根据您调用 map 的集合的静态类型（类型编译器在编译时就知道）。 set 的静态类型是 TreeSet，但 diffset 的静态类型是 Set。两者（运行时）的动态类型都是TreeSet。

当您在 set（TreeSet）上调用 map 时，编译器会选择 immutable.this.SortedSet.canBuildFrom[Int](math .this.Ordering.Int) 作为 CanBuildFrom。

当您在 diffset（Set）上调用 map 时，编译器会选择 immutable.this.Set.canBuildFrom[Int] 作为 CanBuildFrom。

为什么 for 版本最终未排序

循环脱

for (i <- genSet; x = i + 1) {
  println(x)
}

糖为

genSet.map(((i) => {
              val x = i.$plus(1);
              scala.Tuple2(i, x)
            })).foreach(((x$1) => x$1: @scala.unchecked match {
              case scala.Tuple2((i @ _), (x @ _)) => println(x)
            }))

脱糖版本包含一个 map 函数，该函数将使用未排序的 CanBuildFrom，正如我上面所解释的。

另一方面，循环

for (i <- genSet) {
  val x = i + 1
  println(x)
}

脱糖到

genSet.foreach(((i) => {
              val x = i.$plus(1);
              println(x)
            }))

Which 根本不使用 CanBuildFrom，因为没有返回新的集合。

Answer 2

Set 不保证顺序。即使底层类是 TreeSet，如果预期结果是 Set，您也会在第一次转换时失去顺序。

如果您想要排序，请不要使用Set。我建议使用 SortedSet。

Answer 3

更改 genSet 的 sig 以返回 SortedSet

def genSet:SortedSet[Int] = {
  TreeSet(5, 4, 3, 2, 1)
}

这可能是某种错误。我本来希望你的代码也能工作。

我认为 map 是罪魁祸首。这会导致相同的行为：

for (i <- genSet.map(_ + 1)) { println(i) }

并且 for(i <- genSet; x = i + 1) 等于 for(x <- genSet.map({i => i + 1}))

Answer 4

你可以这样做：

scala> for (i <-genSet.view; x = i + 1) println(x)
2
3
4
5
6

虽然，这是一种技巧，当你几个月后查看它时，你可能想知道为什么添加 .view ...