scala – 检查集合包含的元素数量

发布时间：2020-12-16 19:18:36 所属栏目：安全来源：网络整理

导读：我需要编写一个函数,如果一个set(此set是另一个函数的输出)包含1个元素,则返回true,否则它将保留原样. 例如： Set(1)返回特定结果,Set(2,4)按原样返回该集合. 如何检查集合中包含的元素数量？解决方法你只需要一个Set上的size方法： scala Set(1).sizeres0

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

例如：

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

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

解决方法

你只需要一个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
}

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!