如何在Scala中从Set中获取随机元素

Question

对于任何给定的集合，例如

val fruits = Set("apple", "grape", "pear", "banana")

如何从fruits中得到一个随机元素?

许多谢谢。

Answer 1

所以，之前发布的每个答案在空间方面都具有复杂度O(n)，因为它们以某种方式创建了整个集合的副本。下面是一个不需要任何额外复制的解决方案(因此它是"常量空间"):

def random[T](s: Set[T]): T = {
  val n = util.Random.nextInt(s.size)
  s.iterator.drop(n).next
}

Answer 2

转换为Vector并从中获得随机元素

scala> val fruits = Set("apple", "grape", "pear", "banana")
fruits: scala.collection.immutable.Set[String] = Set(apple, grape, pear, banana)
scala> import scala.util.Random
import scala.util.Random
scala> val rnd=new Random
rnd: scala.util.Random = scala.util.Random@31a9253
scala> fruits.toVector(rnd.nextInt(fruits.size))
res8: String = apple

Answer 3

从这个问题的其他答案中得到灵感，我想到了:

private def randomItem[T](items: Traversable[T]): Option[T] = {
  val i = Random.nextInt(items.size)
  items.view(i, i + 1).headOption
}

这不会复制任何东西，如果Set(或其他类型的Traversable)为空，也不会失败，并且一眼就能清楚它的作用。如果您确定Set不为空，则可以使用.head代替headOption，返回T。

Answer 4

可以使用slice直接访问Set的元素。当我在处理一个大小变化的集合时，我使用了这个方法，因此每次将其转换为Vector似乎都有些多余。

val roll = new Random ()
val n = roll nextInt (fruits size)
fruits slice (n, n + 1) last

Answer 5

Solution1

随机方式(import scala.util.Random)

scala>  fruits.toList(Random.nextInt(fruits.size))
res0: java.lang.String = banana

Solution2

Math way (no imports)

scala> fruits.toList((math.random*fruits.size).toInt)
res1: String = banana

Answer 6

   import Scala.util.Random
   val fruits = Set("apple", "grape", "pear", "banana").toVector
   val sz =fruits.size
   val num = Random.nextInt(sz)
   fruits(num)

Answer 7

不将Set转换为有序的集合，但使用zipWithIndex，我们可以为集合中的每个项目赋予索引，

fruits.zipWithIndex
Set((apple,0), (grape,1), (pear,2), (banana,3))

因此对于val rnd = util.Random.nextInt(fruits.size)，

fruits.zipWithIndex.find( _._2 == rnd)
Option[(String, Int)] = Some((banana,3))

给定一个空集合，

Set[String]().zipWithIndex.find( _._2 == 3)
Option[(String, Int)] = None

Answer 8

如果您不介意O(n)解决方案:

import util.Random
// val fruits = Set("apple", "grape", "pear", "banana")
Random.shuffle(fruits).head
// "pear"