Leetcode470 用Rand7()实现Rand10()

题目描述

分析

首先可以有一种极简单的思路，即让使用rand7生成足够多的状态，然后再从这些状态中选取出等概率且不少于10的状态，这样通过在这些状态对应到1~10即可形成rand10。以下为通过rand7+rand7的结果中选取状态的一种表述：

class Solution:
    def rand10(self):
        """
        :rtype: int
        """
        while True:
            a=rand7()
            b=rand7()
            if a>4 and b<4:
                pass
            else:
                return (a+b)%10+1

进阶分析

对于上述分析中的过程可以进一步抽象，从进制的角度去考虑状态的生成，即对于任意的rand函数（如rand n），我们总可以把若干个rand n拼凑在一起然后形成n进制数，从而表示出等概率的足够多的状态来覆盖想要生成的rand函数。
对于此题而言，(rand7()-1)*7+rand7() 就能等概率的表示出1~49共49种状态，然后同样通过取模运算即可求出相应的结果：

class Solution:
    def rand10(self):
        """
        :rtype: int
        """
        num=(rand7()-1)*7+rand7()
        while num>40:
        	num=(rand7()-1)*7+rand7()
		return 1+num%10

优化分析

在上述的情况中，仍然可能出现由于num始终大于40而造成的死循环，因而针对这一问题进一步优化。
根据生成的数字进一步的去产生随机数，且使得要丢弃的数字减少，如第一次(rand7()-1)*7+rand7()那么想要保留的是1-40的状态而想要丢弃的是41- 49的状态；那么第二次据此生成时，则需要把上述的九个状态再次定位一次，(x-40-1)*7+rand7(),那么等概率生成的状态是1-63，所以只需要保留1-60的状态而丢弃61-63的状态，这一过程可以继续做下去，选择(y-60-1)*7+rand7()，那么生成的会是1-21这样就只需要丢弃一个状态即可：

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

拓展

若需要用rand10生成rand7，只需要不断的去生成rand10的随机数直到出现1~7的数字即可。