scala版本的梅森旋转随机数算法

最新推荐文章于 2023-05-23 20:14:50 发布

青竹梦

最新推荐文章于 2023-05-23 20:14:50 发布

阅读量2k

点赞数 1

分类专栏： scala

本文链接：https://blog.csdn.net/rowanhaoa/article/details/42321623

版权

这是一个使用Scala编写的Mersenne Twister随机数生成器类，通过tailrec优化，提供了nextInt和nextFloat两种随机数生成方法，适用于各种随机数需求。

摘要由CSDN通过智能技术生成

package xzxz

import scala.annotation.tailrec

class MersenneTwister(seed: Int) {

  private val SIZE: Int = 624

  private val PERIOD: Int = 397

  private val DIFF: Int = SIZE - PERIOD

  private def isOdd(n: Int): Int = n & 1

  private def m32(n: Int): Int = 0x80000000 & n

  private def l31(n: Int): Int = 0x7FFFFFFF & n

  private val mersenneTwister = new Array[Int](SIZE)

  private var index: Int = 0

  resetSeed(seed)

  private def generateNumbers(): Unit = {

    @tailrec
    def rollMersennerTwister(i: Int): Unit = {
      if (i < DIFF) {
        val y = m32(mersenneTwister(i)) | l31(mersenneTwister(i + 1))
        mersenneTwister(i) = mersenneTwister(i + PERIOD) ^ (y >>> 1) ^ (isOdd(y) * 0x9908b0df)
        rollMersennerTwister(i + 1)
      }
    }

    rollMersennerTwister(0)

    def unroll(i: Int): Unit = {
      val y = m32(mersenneTwister(i)