检查集合包含的元素数量

Question

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

例如:

Set(1)返回特定结果,Set(2,4)按原样返回该集合.

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

Answer 1

你只需要size一个Set上的方法:

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

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

另请参阅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
}