编写一个程序,找出第 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