Swift 中隨機數的使用

發表於2015-10-13

在我們開發的過程中,時不時地需要產生一些隨機數。這裡我們總結一下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() % upper_bound的方式,其中upper_bound指定的是上邊界,如下處理:

不過使用這種方法,在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),如下所示:

而如果想指定區間的最小值(如隨機數區間在[5, 100)),則可以如下處理:

當然,在Swift中也可以使用傳統的C函式rand與random。不過這兩個函式有如下幾個缺點:

  1. 這兩個函式都需要初始種子,通常是以當前時間來確定。
  2. 這兩個函式的上限在RAND_MAX=0X7fffffff(2147483647),是arc4random的一半。
  3. rand函式以有規律的低位迴圈方式實現,更容易預測

我們以rand為例,看看其使用:

64位整型隨機數

在大部分應用中,上面講到的幾個函式已經足夠滿足我們獲取整型隨機數的需求了。不過我們看看它們的函式宣告,可以發現這些函式主要是針對32位整型來操作的。如果我們需要生成一個64位的整型隨機數呢?畢竟現在的新機器都是支援64位的了。

目前貌似沒有現成的函式來生成64位的隨機數,不過jstn在stackoverflow上為我們分享了他的方法。我們一起來看看。

他首先定義了一個泛型函式,如下所示:

這個函式中使用了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位的隨機數生成方法便可以如下實現:

我們來試用一下:

當然jstn還提供了Int64,UInt32,Int32的實現,大家可以腦補一下。

浮點型隨機數

如果需要一個浮點值的隨機數,則可以使用drand48函式,這個函式產生一個[0.0, 1.0]區間中的浮點數。這個函式的返回值是Double型別。其使用如下所示:

記住這個函式是需要先呼叫srand48生成一個種子的初始值。

一個小示例

最近寫了一個隨機鍵盤,需要對0-9這幾個數字做個隨機排序,正好用上了上面的arc4random函式,如下所示:

在閉包中,隨機生成兩個數,比較它們之間的大小,來確定陣列的排序規則。還是挺簡單的。

小結

其實如果翻看一下Swift中關於C函式的API,發現還有許多跟隨機數相關的函式,如arc4random_addrandom,erand48等。上面的只是我們經常用到的一些函式,這幾個函式基本上夠用了。當然,不同場景有不同的需求,我們需要根據實際的需求來選擇合適的函式。

以上的程式碼已上傳到github,地址是Random.playground有需要的可以參考一下。

參考

  1. rand(3) / random(3) / arc4random(3) / et al.
  2. Random Swift
  3. How does one generate a random number in Apple’s Swift language?

相關文章