Golang | Leetcode Golang题解之第470题用Rand7()实现Rand10()-爱伦尔生活百科

Golang | Leetcode Golang题解之第470题用Rand7()实现Rand10()

电脑手机 2024-11-25 11:55:50 515

题目：

Golang | Leetcode Golang题解之第470题用Rand7()实现Rand10()

题解：

func rand10() int {
    for {
        a := rand7()
        b := rand7()
        idx := (a-1)*7 + b
        if idx <= 40 {
            return 1 + (idx-1)%10
        }
        a = idx - 40
        b = rand7()
        // get uniform dist from 1 - 63
        idx = (a-1)*7 + b
        if idx <= 60 {
            return 1 + (idx-1)%10
        }
        a = idx - 60
        b = rand7()
        // get uniform dist from 1 - 21
        idx = (a-1)*7 + b
        if idx <= 20 {
            return 1 + (idx-1)%10
        }
    }
}