Scala可变SortedSet的diff()方法

在Scala中，SortedSet是一个有序的集合。与一般的集合不同，SortedSet可以保证其中的元素是按照特定的顺序排列的。在scala.collection.mutable包中，存在着一个可变SortedSet，它支持add、remove、clone等方法，在此基础上，还存在着一个非常实用的方法——diff()。

阅读更多：Scala 教程

Diff()方法

diff()方法可以用于两个SortedSet之间的计算差集。在具体实现中，SortedSet对象A中除去SortedSet对象B中共有元素，剩下的元素会被放入一个新的SortedSet对象中返回。换句话说，这个方法返回的结果就是SortedSet A 中包含但 SortedSet B 中不包含的元素。

下面是一个例子：

import scala.collection.mutable._

object Example {
   def main(args : Array[String]) {
        // 创建两个可变SortedSet
        var set1 = SortedSet(1, 2, 3, 4)
        var set2 = SortedSet(2, 3, 5, 6)

        // 计算差集
        var diffSet = set1.diff(set2)

        println(diffSet) // 输出：SortedSet(1, 4)
   }
}

在上面的代码中，我们创建了两个SortedSet对象set1和set2，其中set1包含了1~4这几个数字，set2包含了2~6这几个数字。然后我们利用diff()方法计算了它们之间的差集。最后的结果输出为1和4，因为它们是set1中特有的数字。

实现方式

下面我们看一下diff()方法的具体实现。

def diff(that: SortedSet[A]): SortedSet[A]

在Scala标准库中，diff()方法的实现是通过SortedSetFactory.scala文件中的SortedSetOps trait实现的。这个trait内部定义了很多对于SortedSet的操作，包括add()、remove()、clone()、intersect()等方法。其中diff()方法就是通过与intersect()方法一样的思路实现的。

在实现时，diff()方法会遍历调用者SortedSet对象中的每一个元素，同时在给定的SortedSet对象中查找是否存在这个元素。如果存在，就将这个元素从调用者的SortedSet对象中去掉，否则就将这个元素加入到结果集合中。