LeetCode-Python-470. 用 Rand7() 实现 Rand10()

已有方法 rand7 可生成 1 到 7 范围内的均匀随机整数，试写一个方法 rand10 生成 1 到 10 范围内的均匀随机整数。

不要使用系统的 Math.random() 方法。

 

示例 1:

输入: 1
输出: [7]

示例 2:

输入: 2
输出: [8,4]

示例 3:

输入: 3
输出: [8,1,10]

 

提示:

1. rand7 已定义。
2. 传入参数: n 表示 rand10 的调用次数。

思路：

利用已有的随机函数rand7将等概率空间扩大，然后从大的等概率空间中取小范围的数。

rand7 - 1的范围是0 ~ 6，每个数出现概率相等，为1/7

（rand7 - 1 ） * 7 的结果是[0, 7, 14, 21, 28, 35, 42]，每个数字出现的概率相等，为1/7

所以 tmp = (rand7()-1)*7 + rand7()-1，得到的数字刚好能均匀覆盖0 ~ 48，

0 ~ 48 就是得到的大的等概率空间，

取0~39这一部分出来，返回%10 +1的结果就是rand10的答案，

如果tmp >=40，那么重新再算。

class Solution(object):
    def rand10(self):
        """
        :rtype: int
        """
        while True:
            tmp = (rand7()-1)*7 + rand7()-1
            #0 7 14 21 28 35 42
            if tmp < 40:
                return tmp % 10 + 1