已有方法 rand7 可生成 1 到 7 范围内的均匀随机整数,试写一个方法 rand10 生成 1 到 10 范围内的均匀随机整数。
不要使用系统的 Math.random() 方法。
示例 1:
输入: 1
输出: [7]
示例 2:
输入: 2
输出: [8,4]
示例 3:
输入: 3
输出: [8,1,10]
提示:
rand7 已定义。
传入参数: n 表示 rand10 的调用次数。
进阶:
rand7()调用次数的 期望值 是多少 ?
你能否尽量少调用 rand7() ?
方法一:拒绝采样
所谓随机数,就是随机生成一个数,并且所有情况出现的概率相同,本题要求尽可能少的调用rand7,因此我们不能无脑的使用题目中给的随机函数rand7,但是尽可能少是多少呢?1个可以吗。调用一次肯定不行,一次调用顶多出现7种情况,呢我们调用两次试试看,两次的话共49种情况(7*7),我们需要去掉多余的9种情况(41~49),剩下的只要保证每个数出现的概率为4/49即可(PS:为了保证1~40之间每个数出现的概率都为1/49,我们不能简单的相乘,而是采用7进制进行表示)。
其中调用rand7次数期望值的计算可以参考官方题解
class Solution extends SolBase {
public int rand10() {
int row,col,idx;
do {
row=rand7();
col=rand7();
idx=col+(row-1)*7;
}while(idx>40);
return 1+(idx-1)%10;
}
}
方法二:合理使用被拒绝的随机数
该方法官方题解已经解释的非常清楚了,这里不再过多赘述。
class Solution extends SolBase {
public int rand10() {
int a,b,idx;
while(true) {
a=rand7();
b=rand7();
idx=b+(a-1)*7;
if(idx<=40)
return 1+(idx-1)%10;
a=idx-40;
b=rand7();
idx=b+(a-1)*7;
if(idx<=60)
return 1+(idx-1)%10;
a=idx-60;
b=rand7();
idx=b+(a-1)*7;
if(idx<=20)
return 1+(idx-1)%10;
}
}
}