470. 用 Rand7() 实现 Rand10()
给定方法 rand7 可生成 [1,7] 范围内的均匀随机整数,试写一个方法 rand10 生成 [1,10] 范围内的均匀随机整数。
你只能调用 rand7() 且不能调用其他方法。请不要使用系统的 Math.random() 方法。
每个测试用例将有一个内部参数 n,即你实现的函数 rand10() 在测试时将被调用的次数。请注意,这不是传递给 rand10() 的参数。
示例 1:
输入: 1
输出: [2]
示例 2:
输入: 2
输出: [2,8]
示例 3:
输入: 3
输出: [3,8,10]
提示:
1 <= n <= 105
进阶:
rand7()调用次数的 期望值 是多少 ?
你能否尽量少调用 rand7() ?
非常好的题解:解析
# The rand7() API is already defined for you.
# def rand7():
# @return a random integer in the range 1 to 7
class Solution:
# 解析:https://leetcode.cn/problems/implement-rand10-using-rand7/solution/xiang-xi-si-lu-ji-you-hua-si-lu-fen-xi-zhu-xing-ji/
def rand10(self):
"""
:rtype: int
"""
row, col, idx = 0, 0, 0
while True:
row = rand7()
col = rand7()
idx = (row - 1) * 7 + col
if idx <= 40:
return 1 + (idx - 1) % 10
# The rand7() API is already defined for you.
# def rand7():
# @return a random integer in the range 1 to 7
class Solution:
# 解析:https://leetcode.cn/problems/implement-rand10-using-rand7/solution/xiang-xi-si-lu-ji-you-hua-si-lu-fen-xi-zhu-xing-ji/
def rand10(self):
"""
:rtype: int
"""
# row, col, idx = 0, 0, 0
# while True:
# row = rand7()
# col = rand7()
# idx = (row - 1) * 7 + col
# if idx <= 40:
# 返回结果,+1是为了解决 40%10为0的情况
# return 1 + (idx - 1) % 10
while True:
num = (rand7() - 1)*7 + rand7()
# 如果在40以内,那就直接返回
if num <= 40:
return 1 + (num - 1) % 10
# 说明刚刚生成的在41-49之间,利用随机数再操作一遍
num = (num - 40 - 1) * 7 + rand7()
if num <= 60:
return 1 + (num) % 10
# 说明刚刚生成在61-63之间,利用随机数再操作一遍
num = (num - 60 - 1) * 7 + rand7()
if num <= 20:
return 1 + (num) % 10