我需要编写一个函数,如果一个set(这个set是另一个函数的输出)包含1个元素,则返回true,否则它将保留原样.
例如:
Set(1)返回特定结果,Set(2,4)按原样返回该集合.
如何检查集合中包含的元素数量?
你只需要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 }