把只包含因子 2、3 和 5 的数称作丑数(Ugly Number)。例如 6、8 都是丑数,但 14 不是,因为它包含因子 7。习惯上我们把 1 当做是第一个丑数。求按从小到大的顺序的第 N 个丑数。
解题思路
假设我们现在已经有了一个丑数的有序数组,如果要找到下一个丑数,则可以将数组中的每一个数乘以2,
并将其中第一个大于当前丑数的的结果记为M2,同样将当前有序数组每一个数都乘以3,第一个大于当前丑数的的结果记为M3,同样方式得到乘以5的第一个大于当前丑数的结果记为M5。可以下一个丑数必然是min(M2, M3, M5)。
事实上我们并不必要将数组中的每个数都乘以2,3,5。对于乘以2来说,我们只要找到第一个乘以2大于当前丑数的数在数组中的位置,同样找到第一个乘以3,5大于当前丑数的数的位置。如果当前丑数记为M,然后就可以使用min(M2, M3, M*5)来产生下一个丑数。
class Solution:
def GetUglyNumber_Solution(self, index):
# write code here
if index<1:
return 0
vec = [1]
ith2,ith3,ith5=0,0,0
while index>1:
val = min(vec[ith2]*2, min(vec[ith3]*3, vec[ith5]*5))
vec.append(val)
while vec[ith2]*2 <= val:
ith2 += 1
while vec[ith3]*3 <= val:
ith3 += 1
while vec[ith5]*5 <= val:
ith5 += 1
index -= 1
return vec.pop()