转载

总结 Swift 中随机数的使用

总结 Swift 中随机数的使用

在我们开发的过程中,时不时地需要产生一些随机数。这里我们总结一下Swift中常用的一些随机数生成函数。这里我们将在Playground中来做些示例演示。

整型随机数

如果我们想要一个整型的随机数,则可以考虑用arc4random系列函数。我们可以通过man arc4random命令来看一下这个函数的定义:

The arc4random() function uses the key stream generator employed by the arc4 cipher, which uses 8*8 8 bit S-Boxes. The S-Boxes can be inabout (21700) states. The arc4random() function returns pseudo-random numbers in the range of 0 to (232)-1, and therefore has twice the range of rand(3) and random(3).

arc4random使用了arc4密码加密的key stream生成器(请脑补),产生一个[0, 2^32)区间的随机数(注意是左闭右开区间)。这个函数的返回类型是UInt32。如下所示:

arc4random()                // 2,919,646,954

如果我们想生成一个指定范围内的整型随机数,则可以使用arc4random() % upper_bound的方式,其中upper_bound指定的是上边界,如下处理:

arc4random() % 10            // 8

不过使用这种方法,在upper_bound不是2的幂次方时,会产生一个所谓 Modulo bias(模偏差) 的问题。

我们在控制台中通过man arc4random命令,可以查看arc4random的文档,有这么一条:

arc4random_uniform() will return a uniformly distributed random number less than upper_bound. arc4random_uniform() is recommended over constructions like ‘’arc4random() % upper_bound’‘ as it avoids “modulo bias” when the upper bound is not a power of two.

因此可以使用arc4random_uniform,它接受一个UInt32类型的参数,指定随机数区间的上边界upper_bound,该函数生成的随机数范围是[0, upper_bound),如下所示:

arc4random_uniform(10)       // 6

而如果想指定区间的最小值(如随机数区间在[5, 100)),则可以如下处理:

let max: UInt32 = 100 let min: UInt32 = 5 arc4random_uniform(max - min) + min           // 82

当然,在Swift中也可以使用传统的C函数rand与random。不过这两个函数有如下几个缺点:

这两个函数都需要初始种子,通常是以当前时间来确定。

这两个函数的上限在RAND_MAX=0X7fffffff(2147483647),是arc4random的一半。

rand函数以有规律的低位循环方式实现,更容易预测

我们以rand为例,看看其使用:

srand(UInt32(time(nil)))            // 种子,random对应的是srandom rand()                              // 1,314,695,483 rand() % 10                          // 8

64位整型随机数

在大部分应用中,上面讲到的几个函数已经足够满足我们获取整型随机数的需求了。不过我们看看它们的函数声明,可以发现这些函数主要是针对32位整型来操作的。如果我们需要生成一个64位的整型随机数呢?毕竟现在的新机器都是支持64位的了。

目前貌似没有现成的函数来生成64位的随机数,不过jstn在stackoverflow上为我们分享了他的方法。我们一起来看看。

他首先定义了一个泛型函数,如下所示:

func arc4random  (type: T.Type) -> T {     var r: T = 0     arc4random_buf(&r, UInt(sizeof(T)))     return r }

这个函数中使用了arc4random_buf来生成随机数。让我们通过man arc4random_buf来看看这个函数的定义:

arc4random_buf() function fills the region buf of length nbytes with ARC4-derived random data.

这个函数使用ARC4加密的随机数来填充该函数第二个参数指定的长度的缓存区域。因此,如果我们传入的是sizeof(UInt64),该函数便会生成一个随机数来填充8个字节的区域,并返回给r。那么64位的随机数生成方法便可以如下实现:

extension UInt64 {  static func random(lower: UInt64 = min, upper: UInt64 = max) -> UInt64 {   var m: UInt64   let u = upper - lower   var r = arc4random(UInt64)   if u > UInt64(Int64.max) {    m = 1 + ~u   } else {    m = ((max - (u * 2)) + 1) % u   }   while r < m {    r = arc4random(UInt64)   }   return (r % u) + lower  } } 

我们来试用一下:

UInt64.random()                   // 4758246381445086013

当然jstn还提供了Int64,UInt32,Int32的实现,大家可以脑补一下。

浮点型随机数

如果需要一个浮点值的随机数,则可以使用drand48函数,这个函数产生一个[0.0, 1.0]区间中的浮点数。这个函数的返回值是Double类型。其使用如下所示:

srand48(Int(time(nil))) drand48()                       // 0.396464773760275

记住这个函数是需要先调用srand48生成一个种子的初始值。

一个小示例

最近写了一个随机键盘,需要对0-9这几个数字做个随机排序,正好用上了上面的arc4random函数,如下所示:

let arr = ["0", "1", "2", "3", "4", "5", "6", "7", "8", "9"] let numbers = arr.sort { (_, _) -> Bool in     arc4random() < arc4random() }

在闭包中,随机生成两个数,比较它们之间的大小,来确定数组的排序规则。还是挺简单的。

小结

其实如果翻看一下Swift中关于C函数的API,发现还有许多跟随机数相关的函数,如arc4random_addrandom,erand48等。上面的只是我们经常用到的一些函数,这几个函数基本上够用了。当然,不同场景有不同的需求,我们需要根据实际的需求来选择合适的函数。

以上的代码已上传到github,地址是 Random.playground 有需要的可以参考一下。

参考

正文到此结束
Loading...