题目
给定方法 rand7 可生成 [1,7] 范围内的均匀随机整数,试写一个方法 rand10 生成 [1,10] 范围内的均匀随机整数。
你只能调用 rand7() 且不能调用其他方法。请不要使用系统的 Math.random() 方法。
每个测试用例将有一个内部参数 n,即你实现的函数 rand10() 在测试时将被调用的次数。请注意,这不是传递给 rand10() 的参数。
思路
拒绝采样
(rand7() - 1) * 7 + rand7()
可以等概率生成1 ~ 49
的数字
其中(rand7() - 1) * 7
可以取到0 ~ 42
的数字
因为rand7() - 1
能拿到0,1,2,3,4,5,6
再乘 7
后得到集合是{0,7,14,21,28,35,42
}
后面rand7()
得到的集合为1,2,3,4,5,6,7
对应每个加起来,范围就是1 ~ 49
class Soultion extends SolBase{
public int rand10(){
while(true){
int ans = (rand7() - 1) * 7 + rand7();
if(ans >= 1 && ans <= 10)
return ans;
}
}
}
优化:
降低对 rand7
的调用次数,范围[0,49]
中,只有[1,10]
范围内的数据会被接受返回,其余情况均被拒绝重试
为了尽可能少的调用 rand7
方法,可以从 [0,49]
中取与 [1,10]
成倍数关系的数,来进行转换
取 [0,49]
中的 [1,40]
范围内的数来代指 [1,10]
首先在 [0,49]
中取 [1,40]
仍为等概率,其次形如 x1
的数值有 4
个(1、11、21、31
),形如 x2
的数值有 4
个(2、12、22、32
)… 因此最终结果仍为等概率
class Solution extends SolBase {
public int rand10() {
while (true) {
int ans = (rand7() - 1) * 7 + rand7();
if (1 <= ans && ans <= 40)
return ans % 10 + 1;
}
}
}
进一步拓展:(randX() - 1) * Y + randY()
randX() - 1
可以拿到等概率的0,1,2,3,4,... X-1
(randX() - 1) * Y
乘Y拿到的集合为{0,Y,2Y,3Y,4Y,... Y(X-1)
}
randY()
可以拿到1,2,3,4,... Y
最终集合范围为1,2,3,4,... XY
,即当X=7
和Y= 7
时,即可满足1~7*7=49
的范围