Scala TreeSet count() 方法

Scala TreeSet 是一个有序的集合，它使用红黑树实现存储。Scala TreeSet count() 方法用于返回集合中元素的个数。

方法定义

以下是 TreeSet count() 方法的方法定义 -

def count(p: (A) => Boolean): Int

• 参数：
  • p - 接受参数类型为 A 的函数，该函数指定了用于计数的条件。
• 返回值：集合中符合条件的元素的数量。

示例

以下是一个示例程序，用于说明 Scala TreeSet count() 方法的用法。

import scala.collection.immutable.TreeSet

object TreeSetDemo {
   def main(args: Array[String]) {
      val treeSet1: TreeSet[Int] = TreeSet(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
      println("treeSet1: " + treeSet1)

      // 计数元素的值大于 5 的元素数量
      val count1 = treeSet1.count(_ > 5)
      println("Count of elements > 5: " + count1)

      // 计数元素的值小于或等于 5 的元素数量
      val count2 = treeSet1.count(_ <= 5)
      println("Count of elements <= 5: " + count2)

      val treeSet2: TreeSet[String] = TreeSet("Apple", "Banana", "Orange", "Mango")
      println("treeSet2: " + treeSet2)

      // 计数元素的长度大于 5 的元素的数量
      val count3 = treeSet2.count(_.length > 5)
      println("Count of elements with length > 5: " + count3)
   }
}

输出结果如下 -

treeSet1: TreeSet(1, 2, 3, 4, 5, 6, 7, 8, 9, 10)
Count of elements > 5: 5
Count of elements <= 5: 5
treeSet2: TreeSet(Apple, Banana, Mango, Orange)
Count of elements with length > 5: 2

在上面的例子中，我们创建了两个 TreeSet 对象，一个包含整数，另一个包含字符串。我们使用 count() 方法分别计数符合不同条件的元素的数量。在第一个 TreeSet 对象中，我们计数元素的值大于 5 和小于或等于 5 的元素的数量。在第二个 TreeSet 对象中，我们计数元素的长度大于 5 的元素的数量。