rand5()生成rand7()

今天面字节一面遇到这样的题：
给出一个rand5()，来生成一个rand7()
面试官看我没思路，又问我rand7()生成rand5()可以怎样做？
我说rand7()生成1-7的随机数，而rand5()生成1-5的，那么rand7()出来的随机数如果是6-7的话就重新rand7()，直到生成1-5为止。
这样做是没问题，但是从rand5()来生成rand7()，想了几分钟没想出来，遂不再追问
过后我查阅资料之后，发现挺简单的
可以创建一个5*5的二维数组，从1递增到7，最后不足7个数时用0补足（重新摇的标志位）

	vector<vector<int>> v = {
            {1, 2, 3, 4, 5},
            {6, 7, 1, 2, 3},
            {4, 5, 6, 7, 1},
            {2, 3, 4, 5, 6},
            {7, 0, 0, 0, 0}
    };

横坐标的取值范围为[0-4]，每个概率为1/5
纵坐标的取值范围为[0-4]，每个概率为1/5
那么25个数里面每个数被访问的概率为1/25，如果为0则重新摇
具体代码如下：

#include <iostream>
#include <vector>
#include <cmath>
using namespace std;

int rand5()
{
    return rand()%5+1;
}

int rand7()
{
    vector<vector<int>> v = {
            {1, 2, 3, 4, 5},
            {6, 7, 1, 2, 3},
            {4, 5, 6, 7, 1},
            {2, 3, 4, 5, 6},
            {7, 0, 0, 0, 0}
    };

    int res = 0;
    while(res == 0)
    {
        int i = rand5();
        int j = rand5();
        res = v[i-1][j-1];
    }

    return res;
}

int main()
{
    int size = 1000000;
    vector<int> v(10);
    for(int i = 0; i < size; i++){
        v[rand7()]++;
    }
    for(int i = 1; i <= 7; i++){
        cout<<1.0*v[i]/size<<endl;
    }

    return 0;
}

用100w来验证一下，可以看到几乎是相等的

能否不用数组来做呢？
可以先构造大的整数区间：rand5*(rand5-1)
rand5-1的取值范围为：0-4
rand5*(rand5-1)的取值为：0，5，10，15，20
再填充整数区间：rand5*(rand5-1)+rand5
rand5的范围为1-5，加上前面的数则[1-25]
最后做映射：因为7*3=21，最接近25，则超过21则重新摇，剩下的[1-21]%7

int rand7()
{
    int i;
    do{
        i = 5*(rand5()-1)+rand5();
    }while(i > 21);

    return i%7+1;
}

验证

总结：
如果用randA来生成randB（A < B）
可以套用上面的模板：randB = A *（randA - 1）+ randA

参考：
https://www.cnblogs.com/dwdxdy/archive/2012/07/28/2613135.html