我的技术点滴总结：一种加权随机抽奖算法

沉下心来学鲁班

已于 2023-09-16 06:55:01 修改

阅读量1.5k

点赞数 1

分类专栏：实践案例文章标签：算法 golang 后端

于 2023-09-16 06:36:50 首次发布

本文链接：https://blog.csdn.net/xiaojia1001/article/details/132914175

版权

实践案例专栏收录该内容

25 篇文章

订阅专栏

本文介绍了如何设计一种公平的抽奖算法，通过按中奖次数衰减权重来降低已中奖者再次中奖的概率，同时讨论了如何改进算法以处理大量用户和多轮抽奖的情况，最后提供了封装的工具类实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1.引言

一场活动少不了抽奖，抽奖的方式有很多：

循点私，可以在指定的用户范围内抽奖；
只为走形式，可以直接让指定人中奖；
如果要公平点，可以来个随机抽奖；
……

循私的事我们就不去宏扬了，还是介绍一种比较公平的抽奖算法，毕竟这种涉及到利益的事，公平才是最难的。

抽奖中有一个比较典型的问题：重复中奖。

抽奖往往不只一轮，在多轮抽奖下，如果之前中过奖的人接连继续中奖，这场活动会瞬间引来大家非议，主办方也会陷入尴尬，为避免把一场活动办砸，我们有必要解决下这个问题。

针对这个问题，从用户层面，一般有两种做法：

不能重复中奖：粗暴直接，将中过奖的人都排除掉；
降低概率：允许重复中奖，但要降低已中奖人的再次中奖概率；

第一种比较简单，我们今天主要讨论第二种。

2.思路

要降低已中奖者的中奖概率，至少要满足两点诉求：

对于已中奖者，再次中奖概率要低于未中奖者；
对于中奖次数多的用户，中奖概率要低于中奖次数少的用户；

一个有效的办法是：按中奖次数给相应用户降低权重。

2.1 降权算法

假设每个人初始权重值都是20，现在有A、B、C、D、E共5个人，分别中过4、3、2、1、0次奖，我们可以让权重值按照每个人中过奖的次数按指数衰减，衰减后每个人的权重变化如下：

A：20 >> 4 = 1 （用向右移4位来表示权重衰减4次，下面同理）
B：20 >> 3 = 2
C: 20 >> 2 = 5
D: 20 >> 1 = 10
E: 20 >> 0 = 20

这里采用指数方式是为了让衰减效果更好，可以看出中奖次数越多，权重衰减得越明显。

如果用代码实现权重计算，示例并说明如下：

winCount：表示用户列表+中奖次数数据，其中数组下标表示用户，数组元素表示用户的中奖次数；
当权重衰减到小于1时，默认为1，以保证极端情况下每个用户仍然有中奖概率；
return weights: 计算好的用户权重数据；

func calculateWeights(winCount []int) []int {
	weights := make([]int, len(winCount))
	for i, count := range winCount {
		weight := 20 >> count    // 降权
		if weight < 1 {
			weight = 1
		}
		weights[i] = weight
	}
	return weights
}

2.2 加权抽奖

为了更好理解，我们先看下最简单的抽奖做法，在0到n（用户数量）之间取随机数，抽出的随机数就是中奖者，每个用户的中奖机会相同。

func draw() string {
	users := []string{"A", "B", "C", "D", "E"}
	rand.Seed(time.Now().UnixNano()) // 随机数种子
	index := rand.Intn(len(users))
	return users[index]
}

如果加入权重，每个用户的中奖机会则是不同的，比较直观的做法就是按照权重在用户列表中重复添加一定数量的用户，例如：

E的权重是20，则用户列表中重复20个E
D的权重是10，则用户列表中重复10个D
C的权重是5，则用户列表中重复5个C
B的权重是2，则用户列表中重复2个B
A的权重是1，则用户列表中重复1个A（等于没有重复）

最终加入权重后的用户列表如下：
[权重对抽取结果影响的示意图]

基于这个用户列表作随机算法抽取，显然中奖概率是：E > D > C > B > A，这样就产生了加权的效果。

2.3 算法改进

上面重复添加用户的作法虽然实现了效果，但是会有个问题：权重用户列表会随着权重值以几何倍数增长。原因如下：

上面示意的5个用户只是为了说明方便，实际场景可能会有成千上万的用户，甚至更多；
当权重值很大时，权重用户列表可能会非常大；
假设有10000用户，权重值1000，则用户列表需要从10000扩充到一千万；
列表无限制增长，会给内存和程序的稳定性带来风险；

基于此，我们还需要对算法作一些改进。

有一种做法是将权重值叠加，计算一个总权重数，在总权重范围内做随机数抽取，最终看随机数落在哪个用户的权重区间。还以上面的例子说明：

A的权重是1，则A的权重区间是[0-0]
B的权重是2，则B的权重区间是[1-2];
C的权重是5，则B的权重区间是[3-7];
D的权重是10，则B的权重区间是[8-17];
E的权重是20，则B的权重区间是[18-37];
A、B、C、D、E加起来的总权重是38，在38之内抽取随机数，其实有效抽取区间是[0-37]
假设如果抽到的随机数是5，则落在C的权重区间，即C中奖。

用代码表示的话大概如下：

入参userWeights数组表示用户列表及每个用户的权重，仍然用数组下标表示用户，数组元素值表示权重（这里假设已经计算好权重）;
返回值win表示抽中的用户，即用户在userWeights中对应的下标；

func draw(userWeights []int) (win int) {
	// 计算总权重
	var totalWeight int
	for _, weight := range userWeights {
		totalWeight += weight
	}
	
	// 在总权重范围内随机抽取一个权重值 n
	rand.Seed(time.Now().UnixNano())
	n := rand.Intn(totalWeight)
	
	// 遍历用户列表，将n与每个用户的权重做减法，当n第一次小于0时，就表示n落在这个用户的权重区间内，此用户中奖
	for i, weight := range userWeights {
		n -= weight
		if n < 0 {
            win = i
            break
		}
	}
	return
}

通过此番改造后，算法性能只和用户列表大小有关，与默认权重不再有关系。

2.4 支持同时抽多个用户

年会抽奖一轮往往不止抽一个用户，如果要抽取多个用户，上面的实现还需要稍微改造下。

首先入参需要扩展一个参数num, 表示一次抽取的用户数；
开启一个循环，多次运行上面的抽奖算法；
还需要处理一个问题：循环中的前后两次有可能抽中同一用户，处理方式如下：
- 如果抽到的用户之前已经被抽中过，则跳过此用户，相当于把中奖机会平移给后面未中奖的用户；
- 总权重要减掉已经中过奖的用户，以保证下一次抽奖抽到的随机数n能落在有效用户身上。

func draw(userWeights []int, num int) []int {
    winned := make([]int, 0, num)
    ……   // 总权重计算省略
    
    // 多次抽取
	for i := 0; i < num; i++ {
		n := this.random.Intn(totalWeight)
		for j, weight := range userWeights {
			if contains(winned, j) {   // 已经中过奖的用户直接跳过，contains用于检查数组中是否包含一个元素，代码省略
				continue
			}
			n -= weight
			if n < 0 {
				totalWeight -= weight  // 总权重totoalWeight要减掉已经中奖的用户权重
				winned = append(winned, j)
				break
			}
		}
	}
	return winned
}

3. 提取工具库

为方便使用，我们可以封装一个工具类，这个工具类定义如下：

type WeightedRandom struct {
	random        *rand.Rand // 随机数生成器
	defaultWeight int            // 用户初始权重
}

它需要有一个构造方法来初始化随机数生成器，并支持调用方指定初始权重。

func NewWeightedRandom(defaultWeight int) *WeightedRandom {
	return &WeightedRandom{
		random:        rand.New(rand.NewSource(time.Now().UnixNano())),
		defaultWeight: defaultWeight,
	}
}

它还需要有一个公开的API方法提供抽奖功能，这个方法只需要做两件事，直接调用我们之前封装好的两个方法就行：

根据已中奖次数计算权重；
根据权重来抽取一定数量的用户；

func (this *WeightedRandom) Draw(winCount []int, num int) []int {
    userWeights := this.calculateWeights(winCount)
	return this.draw(userWeights, num)
}

关于初始权重defaultWeight，稍微说明下，这个参数决定了中奖次数对权重影响的细腻程度；