原理
微信红包的生成算法的本质问题是:怎么能随机生成m个数,让其和等于n?如10元分9包就是随机生成9个数字,使其和等于10.
“剪绳子”法:把一根10米长的绳子剪为9段,第一段的“剪点”是第一段绳子的长度,第二段的“剪点”是前2段绳子长度总和,第三段的“剪点”是前3段绳子长度总和,依次类推……因此,我们可以创建一个数组:
- 第一个元素为0,最后一个元素是10,
- 为其他9个元素元素随机生成一个(0,10)区间的数,
- 对数组进行排序,
- 数组相邻数字相减所得差值就是我们所要的
实现
上述原理中,在第2步生成随机数可能一样,因此为了避免有人拿到0元空红包,可以给每人预留一分钱。实现如下:
static float[] generateWechatRedEnvelope(int totalMoney, int personCount) {
float[] ret = new float[personCount];
ret[0] = 0;
int i = 1;
//以分转为单位
//下面不考虑随机数的唯一性,给每个人都预留1分钱
int totalByHundred = totalMoney * 100 - personCount;
Random random = new Random();
while (i < personCount) {
ret[i] = random.nextInt(totalByHundred);
i++;
}
Arrays.sort(ret);
i = 0;
while (i < personCount - 1) {
ret[i] = (ret[i + 1] - ret[i] + 1) / 100f;//+1:每个人都预留1分钱
i++;
}
ret[i] = (totalByHundred - ret[i] + 1) / 100f;//+1:每个人都预留1分钱
return ret;
}
效率
这种实现的时间复杂度主要在排序,为O(nlogn),效率并不高