rand java,面试题：使用rand5()生成rand7()

前言

读研究生这3年，思维与本科相比变化挺大的，这几年除了看论文、设计方案，更重要的是学会注重先思考、再实现，感觉更加成熟吧，不再像个小P孩，人年轻时总会心高气傲。有1道面试题：给你一个能生成1到5随机数的函数，用它写一个函数生成1到7的随机数，即使用函数rand5()来实现函数rand7()。本文记录一下现在的我的解题过程。

关键点

直接用rand5()实现rand7()好像不好做，那么反过来了？直接利用rand7()实现rand5()效果如何，伪代码直接设计如下：

int rand5(){

int x = 0;

do {

x = rand7();

} while(x > 5);

return x;

}

对于rand7()方法[1, 7]的数是等概率生成的，而我们把[1, 5]范围外的数直接丢弃，每一次循环被丢弃的概率为2/7，我们来分析返回x = 1的概率，有如下公式

Pr[x = 1] = 1/7 + (2/7) * 1/7 + (2/7)^2 * 1/7 + (2/7)^3 * 1/7 + ...

=1/7 * (1 + 2/7 + (2/7)^2 + ...) // 等比数列

=1/7 * 1 / (1 - 2/7)

=1/7 * 7/5

=1/5

即rand5()方法生成1的概率为1/5，同理，生成2、3、4、5的概率为1/5。我们可以留意到，若X ≥ Y，那么rand_X()一定可以用来实现rand_Y()。我们要实现rand7()，必须找到大于等于7的数X。

方案实现

我们现在手头只有rand5()，怎么实现randX()呢？一个直觉是直接实现rand25()，伪代码设计如下

int rand25() {

return 5 * (rand5() - 1) + rand5();

}

有了25≥7，我们将rand7()实现如下

int rand7() {

int x = 0;

do {

x = rand25();

} while(x > 7);

return x;

}

优化

当然，我们可以利用倍数做进一步优化

int rand7() {

int x = 0;

do {

x = rand25();

} while(x > 21);

return x / 3;

}

转载至链接:https://my.oschina.net/keyven/blog/1831712