剑指 Offer 49. 丑数-Python题解

编写一个程序，找出第 n 个丑数。

丑数就是质因数只包含 2, 3, 5 的正整数。

示例:

输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:

1 是丑数。
n 不超过1690。

class Solution:
    def nthUglyNumber(self, n: int) -> int:
        ugly_num_list = [1]
        
        count = 1
        #  丑数,把只包含2,3,5的数称为丑数,其实我第一看没太懂
        # 翻译以下就是 一个数如果能分解成2,3,5中其中一个或多个相乘得到,既是丑数,比如8,可由 2*2*2得到,所以是丑数
        # 那我们可以得到什么呢?其实我们就从小到大组装丑数,存储到列表中就好,
        # 第一个丑数是1,那么我们从1,开始乘以2,3,5,其实我们是使用了空间换时间,存储到列表中,达到了指定的排位的丑数,就直接返回.
        # 下面的这几个索引很巧妙,分别代表了如今2,3,5的最小下标,因为我们要从小到大排序丑数,那么我们要求出下一位的比当前位大的最少的丑数
        # 举例: 
        # 从头开始,1是第一位丑数,索引对应为0,那么下一位是什么呢? 
        # ugly_num_list[0]*2 ugly_num_list[0]*3 ugly_num_list[0]*5 ,既是 1*2,1*3,1*5
        # 都是丑数,下一位是三者的最小值,就是2,丑数2对应的索引为1,而two_index变成了1,同理
        # 下一次, ugly_num_list[1]*2 ugly_num_list[0]*3 ugly_num_list[0]*5 ,既是 2*2,1*3,1*5
        # 这次最小的丑数是3,则three_index变成了1
        # 同理,再下一个丑数,从这几个丑数的最小的那个获取 
        # ugly_num_list[1]*2 ugly_num_list[1]*3 ugly_num_list[0]*5 ,既是 2*2,2*3,1*5
        # 这次最小的丑数是4,two_index索引值变成了2,以此类推
        two_index = 0
        three_index = 0
        five_index = 0

        res = 1
        while count < n :
            count += 1
            res = min(2 * ugly_num_list[two_index], 3 * ugly_num_list[three_index], 5 * ugly_num_list[five_index])
            if res == 2 * ugly_num_list[two_index]:
                two_index += 1
            if res == 3 * ugly_num_list[three_index]:
                three_index += 1
            if res == 5 * ugly_num_list[five_index]:
                five_index += 1
            ugly_num_list.append(res)
        # print(ugly_num_list)
        return res