Loading... ## 介绍 首先,什么是加权随机?当我们从某种容器中随机选择一个元素,每个元素被选中的机会并不相等,而是由相对“权重”(或概率)被选中的,也就是说我们想要有“偏心”的得到某种随机结果。举一个例子,假如现在有一个权重数组 w = {1, 2, 4, 8},它们代表如下规则。 - $\frac{1}{(1+2+4+8)} = \frac{1}{15} \approx 6.6$ \% 的机会选中索引 0 - $\frac{2}{(1+2+4+8)} = \frac{2}{15} \approx 13.3$ \% 的机会选中索引 1 - $\frac{3}{(1+2+4+8)} = \frac{4}{15} \approx 26.6$ \% 的机会选中索引 2 - $\frac{1}{(1+2+4+8)} = \frac{8}{15} \approx 53.3$ \% 的机会选中索引 3 在游戏开发的过程中,很多场景都会用到加权随机。例如游戏中的抽奖,我们有 50% 的几率获得金币、40% 的几率获得钻石、9% 的几率获得普通装备,1% 的几率获得极品装备。 再比如 nginx 的配置中,也有权重配置。 ## 解决方案 ### 方案一 第一个方法是在我们的候选列表中,包含了基于权重的每个索引的预期数量,然后从该列表中随机选择。 假设现在有权重列表 {1, 2, 4, 8},那我们得到的候选列表将是 {0, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3}。 然后通过 rand.Intn() ,获取一个随机数,就完成了,代码如下。 ```go func weightedRandomS1(weights []int) int { if len(weights) == 0 { return 0 } var indexList []int for i, weight := range weights { cnt := 0 for weight > cnt { indexList = append(indexList, i) cnt++ } } rand.Seed(time.Now().UnixNano()) return indexList[rand.Intn(len(indexList))] } ``` ### 方案二 使用方案一,当权重特别大的时候,这种方案显然效率不高,会浪费很多时间来生成列表,并占用太多的内存。 方案一中的列表不是必须的,方案二避免生成大的列表。由于总权重为 15(1+2+4+8),我们可以生成一个 [0,15) 的随机整数,然后根据这个数字返回索引。代码如下。 ```go func weightedRandomS2() int { rand.Seed(time.Now().UnixNano()) r := rand.Intn(15) if r <= 1 { return 0 } else if 1 < r && r <= 3 { return 1 } else if 3 < r && r <= 7 { return 2 } else { return 3 } } ``` ### 方案三 方案二避免了方案一中的生成列表,因此效率更高了。但是我们必须写很多的 if else 代码,这看起来太难看了,为了避免编写过多的 if else 代码,衍生出了方案三。 不必将 r 与所有的范围进行比较。我们可以依次减去总权重,任何时候结果小于等于零,我们就可以返回它。这种方法可以叫做放弃临时名单。 ```go func weightedRandomS3(weights []int) int { rand.Seed(time.Now().UnixNano()) r := rand.Intn(15) for i, v := range weights { r = r - v if r <= 0 { return i } } return len(weights) - 1 } ``` ### 方案四 对于方案三,r 小于等于 0 的速度越快,我们的算法就越高效。那么我们如何让 r 到达 0 更快呢? 直观感受上,如果 r 减去最大的权重,就会更快到达 0 ,所以在运行 weightedRandom 前,我们可以对 weights 按照权重从大到小排序。 ```go func weightedRandomS4(weights []int) int { sort.Sort(sort.Reverse(sort.IntSlice(weights))) rand.Seed(time.Now().UnixNano()) r := rand.Intn(15) for i, v := range weights { r = r - v if r <= 0 { return i } } return len(weights) - 1 } ``` 可以通过数学期望来证明我们的想法。 **最佳顺序** `{8, 4, 2, 1}` E = $\frac{8}{15}$ $ \times $1 + $\frac{4}{15}$ $ \times $2 + $\frac{2}{15}$ $ \times $3 + $\frac{1}{15}$ $ \times $4 = $\frac{16}{10}$ **相对最佳顺序,较差的顺序** `{2, 4, 8, 1}` E = $\frac{2}{15}$ $ \times $1 + $\frac{4}{15}$ $ \times $2 + $\frac{8}{15}$ $ \times $3 + $\frac{1}{15}$ $ \times $4 = $\frac{24}{10}$ **最差的顺序** `{1, 2, 4, 8}` E = $\frac{1}{15}$ $ \times $1 + $\frac{2}{15}$ $ \times $2 + $\frac{4}{15}$ $ \times $3 + $\frac{8}{15}$ $ \times $4 = $\frac{32}{10}$ 可以看到,最佳顺序,即权重从大到小的排序。 ### 方案五 方案四中,实际上引入了一个新的耗时步骤,我们必须对 weightedRandom 排序,当这是一个很大的列表时,效率也就被拉低了。 在方案五中,我们考虑使用累积权重,而不是原始权重。并且由于累积权重是升序排序的,我们可以使用二分来加快速度,因为二分查找可以将时间复杂度从 $ O(n) $ 变为 $ O(log(n)) $。 ```go func weightedRandomS5(weights []int) int { rand.Seed(time.Now().UnixNano()) sum := 0 var sumWeight []int for _, v := range weights { sum += v sumWeight = append(sumWeight, sum) } r := rand.Intn(sum) idx := sort.SearchInts(sumWeight, r) return weights[idx] } ``` ### 方案六 到目前位置,我们的解决方案已经足够好了,但是仍然有改进的余地。 方案五中,我们使用了 go 标准库的二分查找算法 sort.SearchInts() ,它这是封装了通用的 sort.Search() 函数,如下。 ![sort.SearchInts](https://weizicoding-1318098637.cos.ap-chengdu.myqcloud.com/%E9%9A%8F%E6%9C%BA%E5%8A%A0%E6%9D%83/picture_01.png) sort.Search() 的函数参数需要一个闭包函数,并且这个闭包函数是在 for 循环中使用的,如下。 ![sort.Search](https://weizicoding-1318098637.cos.ap-chengdu.myqcloud.com/%E9%9A%8F%E6%9C%BA%E5%8A%A0%E6%9D%83/picture_02.png) 所以目前无法被编译器正确地内联,从而导致了非实质性的性能开销,在方案六中,我们可以编写一个手动内联的版本。 ```go func weightedRandomS6(weights []int) int { rand.Seed(time.Now().UnixNano()) sum := 0 var sumWeight []int for _, v := range weights { sum += v sumWeight = append(sumWeight, sum) } r := rand.Intn(sum) idx := searchInts(sumWeight, r) return weights[idx] } func searchInts(a []int, x int) int { i, j := 0, len(a) for i < j { h := int(uint(i+j) >> 1) if a[h] < x { i = h + 1 } else { j = h } } return i } ``` 通过基准测试可以看到吞吐量提升了 33% 以上。对于大型数据集,优势越明显。 ![优化前](https://weizicoding-1318098637.cos.ap-chengdu.myqcloud.com/%E9%9A%8F%E6%9C%BA%E5%8A%A0%E6%9D%83/picture_03.png) ![优化后](https://weizicoding-1318098637.cos.ap-chengdu.myqcloud.com/%E9%9A%8F%E6%9C%BA%E5%8A%A0%E6%9D%83/picture_04.png) ### 方案七 目前为止我们所有的方案都有一个共同点 —— 生成一个介于 0 和权重之和之间的随机数,并找出它属于哪个“切片”。 还有一种不同的方法。 ```go func weightedRandomS7(weights []float64) int { var sum float64 var winner int rand.Seed(time.Now().UnixNano()) for i, v := range weights { sum += v f := rand.Float64() if f*sum < v { winner = i } } return winner } ``` 这个算法的一个有趣的特性是你不需要提前知道权重的数量就可以使用它。所以说,它或许可以用于某种流。 尽管这种方案很酷,但它比其他方案慢得多。相对于方案一,它也快了 25% 。 ## 源代码 [https://github.com/guowei-gong/weighted-random](https://github.com/guowei-gong/weighted-random) 最后修改:2024 年 11 月 22 日 © 允许规范转载 赞 如果觉得我的文章对你有用,请随意赞赏