Scala:在范围1到45中模拟乐透号码生成器

时间:2015-03-13 19:19:24

标签: scala scala-collections

以下是从1到45范围内模拟数字彩票的必要解决方案,每当我们生成数字n1时,该数字将从可能的数字集中删除。

是否有可能以更实用的方式实现相同目标?即使用地图,过滤等

def getNumbers :Array[Int] = {

        val range = 1 to 45
        var set = range.toSet

        var resultSet:Set[Int] = Set()
        var current: Int = 0

        while(resultSet.size <  5 ){
            current = Random.shuffle(set).head // pick the head of the shuffled set
            set -= current
            resultSet += current
        }

        resultSet.toArray
    }

“编辑”

示例从范围1到5中选择3个数字

Original Set is {1,2,3,4,5}
{1,2,3,4,5} shuffle(1) picked at random 3
{1,2,4,5}     shuffle(2) picked at random 2
{1,4,5}       shuffle(3) picked at random 4
original Set becomes {1,5}

numbers picked {3,2,4}

每次shuffle随机化一个不同的SET! =&GT;不同的概率

我希望看到一个功能性的“方法”,其中包含5次shuffle而不是1次shuffle!

4 个答案:

答案 0 :(得分:7)

当然,这是可能的。集合API拥有您需要的一切。您正在寻找的是take,它将采用集合的第一个n元素,或者如果小于n,则采用与集合相同的元素。< / p>

Random.shuffle(1 to 45).take(5).toArray

答案 1 :(得分:1)

我和m-z在一起,但如果你真的想要一个不断重新洗牌的功能形式,那么你想要这样的东西:

import scala.util.Random
import scala.annotation.tailrec

val initialSet = 1 to 45

def lottery(initialSet: Seq[Int], numbersPicked: Int): Set[Int] = {
  @tailrec
  def internalTailRec(setToUse: Seq[Int], picksLeft: Int, selection: Set[Int]):Set[Int]= {
    if(picksLeft == 0) selection
    else {
      val selected = Random.shuffle(setToUse).head
      internalTailRec(setToUse.filter(_ != selected), picksLeft - 1, selection ++ Set(selected))
    }
  }
  internalTailRec(initialSet, numbersPicked, Set())
}

lottery(initialSet, 5)

答案 2 :(得分:0)

我更喜欢@ m-z的解决方案,并同意他关于概率的推理。阅读Random.shuffle的来源可能是值得的。

每次迭代都会从range中删除一个元素并将其添加到累加器acc,这与您的命令式方法类似,只是我们不会改变集合和计数器。

import scala.util.Random._
import scala.annotation.tailrec

def getNumbers(): Array[Int] = {
  val range = (1 to 45).toSeq
  @tailrec
  def getNumbersR(range: Seq[Int], acc: Array[Int], i: Int): Array[Int] = (i, range(nextInt(range.size))) match{
    case (i, x) if i < 5 => getNumbersR(range.filter(_ != x), x +: acc, i + 1)
    case (i, x)          => acc
  }
  getNumbersR(range, Array[Int](), 0)
}

scala> getNumbers
res78: Array[Int] = Array(4, 36, 41, 20, 14)

答案 3 :(得分:0)

这将模拟您想要的行为:

画一张,剩余的重新洗牌数据,画一张等等。

import scala.util.Random
import scala.annotation.tailrec

def draw(count: Int, data: Set[Int]): Set[Int] = {

  @tailrec
  def drawRec( accum: Set[Int] ) : Set[Int] =  
    if (accum.size == count ) 
      accum
    else 
      drawRec( accum + Random.shuffle( (data -- accum).toList ).head )

  drawRec( Set() )      
}

实施例

scala> draw(5, (1 to 45).toSet)
res15: Set[Int] = Set(1, 41, 45, 17, 22)

scala> draw(5, (1 to 45).toSet)
res16: Set[Int] = Set(5, 24, 1, 6, 28)