把只包含因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
基本思路:创建数组保存已经找到的丑数,新加入的丑数肯定是某个已加入丑数的2或3或5倍。关键在于,如何保证数组里的丑数是有序的。假设当前数组的最大值为M,T2,T3,T5分别标记乘以相应的倍数后第一个大于M的位置,取T2 * 2,T3 * 3,T5 * 5中的最小值加入当前数组,并及时更新T2,T3,T5。
# -*- coding:utf-8 -*-
class Solution:
def GetUglyNumber_Solution(self, index):
# write code here
array = [1]
p2, p3, p5 = 0, 0, 0
if index == 0:
return 0
else:
while index > len(array):
cur = min(array[p2] * 2, array[p3] * 3, array[p5] * 5)
array.append(cur)
while array[p2] * 2 <= array[-1]:
p2 += 1
while array[p3] * 3 <= array[-1]:
p3 += 1
while array[p5] * 5 <= array[-1]:
p5 += 1
return array[index - 1]