scala - 检查一组包含多少个元素

Question

我需要编写一个函数，如果一个集合(该集合是另一个函数的输出)包含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的大小，并根据该大小返回一个值。例如，我假设如果集合的大小为1，我们需要返回一个特殊值(一个包含值99的集合)，但只要将其替换为实际需要返回的任何特定结果即可。

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