检查集合包含的元素数量

Question

我需要编写一个函数，如果一个set（这个set是另一个函数的输出）包含1个元素，则返回true，否则它将保留原样。

例如：

Set（1）返回一个特定的结果，Set（2,4）按原样返回该组。

如何查看集合中包含的元素数量？

Answer 1

您只需要在Set上使用size方法：

scala> Set(1).size
res0: Int = 1

scala> Set(1,2).size
res1: Int = 2

另请参阅documentation for Set。

假设您的其他功能名为getSet。所以你需要做的就是调用它，然后检查生成的Set的大小，并根据该大小返回一个值。例如，我假设如果set的大小为1，我们需要返回一个特殊值（一个包含值99的Set） - 但只需将其替换为您实际需要返回的任何特定结果。

def mySet = {
  val myset = getSet()
  if (myset.size == 1) Set(99)  // return special value
  else myset  // return original set unchanged
}