Scala TreeSet +()方法及示例

阅读更多：Scala 教程

简介

Scala 是一门基于JVM的高级编程语言，它整合了面向对象和函数式编程的特性。Scala 提供了很多集合类，其中 TreeSet 是基于红黑树实现的有序集合。本文将介绍 Scala TreeSet +() 方法的用法及示例。

TreeSet +() 方法

Scala TreeSet 的 +() 方法用于将一个元素添加到 TreeSet 中。该方法的语法如下：

def + (elem: A): TreeSet[A]

其中，elem 表示要添加的元素，TreeSet[A] 表示添加元素后得到的新的 TreeSet 集合。

+() 方法因为会将集合重新排序，所以性能上不如可变集合的 +=() 方法。在集合中添加一批元素时，可变集合的性能更加出色。

TreeSet +() 方法示例

下面示例中，我们将会对一个 TreeSet 集合进行多次元素的添加，并观察 TreeSet 在新元素加入后的顺序。

import scala.collection.immutable.TreeSet

object Main extends App {
  var treeSet = TreeSet(3, 7, 1, 5)

  println(s"初始 TreeSet : treeSet")

  treeSet = treeSet + 8
  println(s"加入元素 8 :treeSet")

  treeSet = treeSet + 0
  println(s"加入元素 0 : treeSet")

  treeSet = treeSet + 6
  println(s"加入元素 6 :treeSet")  
}

输出：

初始 TreeSet : TreeSet(1, 3, 5, 7)
加入元素 8 : TreeSet(1, 3, 5, 7, 8)
加入元素 0 : TreeSet(0, 1, 3, 5, 7, 8)
加入元素 6 : TreeSet(0, 1, 3, 5, 6, 7, 8)

从输出结果来看，每次加入元素后 TreeSet 都会重新进行排序。

TreeSet +() 方法注意事项

由于 TreeSet 是有序集合，相同元素只能存在一个，多次添加相同元素时，只有第一次添加会成功。

import scala.collection.immutable.TreeSet

object Main extends App {
  var treeSet = TreeSet(1, 2, 3, 3, 4)
  println(s"初始 TreeSet : treeSet")

  treeSet = treeSet + 3
  println(s"添加 3 的结果为 :treeSet")

  treeSet = treeSet + 5
  println(s"添加 5 的结果为 : $treeSet")
}

输出：

初始 TreeSet : TreeSet(1, 2, 3, 4)
添加 3 的结果为 : TreeSet(1, 2, 3, 4)
添加 5 的结果为 : TreeSet(1, 2, 3, 4, 5)

从输出结果中可以看出，第二次添加元素 3 并未改变 TreeSet 集合的内容。

结论

Scala TreeSet 的 +() 方法实现了有序集合的元素添加，但是由于排序的影响，性能不如可变集合的 +=() 方法。在添加单个元素时使用 +() 方法可以更好的享受到 TreeSet 的排序特性。同时在添加多个元素时可选择使用可变集合进行优化。