leetcode_470 用 Rand7() 实现 Rand10()

154 阅读1分钟

要求

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

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

 

示例 1:

输入: 1
输出: [7]

示例 2:

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

示例 3:

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

提示:

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

核心代码

# The rand7() API is already defined for you.
# def rand7():
# @return a random integer in the range 1 to 7

class Solution:
    def rand10(self):
        """
        :rtype: int
        """
        while True:
            tmp = (rand7()-1) * 7 + rand7() - 1
            if tmp < 40:
                return tmp % 10 + 1  

image.png

解题思路:我们的(rand7()-1) 的范围是0 ~ 6,(rand7()-1) * 7的范围是0、7、14、21、28、35、42,(rand7()-1) * 7 + (rand7()-1) 就是我们的数据能覆盖到的0 ~ 48,我们只取039的部分,对10取余09,在加上1,范围就是1~10,就是rand10()