49-丑数

题目:我们把只包含因子2、3和5的数称作为丑数。求按从小到大的顺序的第1500个丑数,习惯上我们把1作为第一个丑数。

def ugly_num(n):
    if n<1:
        return 0
    res = [0,1,2,3,4,5]
    i = 6
    t2, t3, t5 = 1, 1, 1
    while i<=n:
        while res[t2]*2<=res[-1]:
            t2 += 1
        while res[t3]*3<=res[-1]:
            t3 += 1
        while res[t5]*5<=res[-1]:
            t5 += 1

        num = min([res[t2]*2,res[t3]*3,res[t5]*5])
 
        res.append(num)

        i += 1

    return res[n]

  注:为了利用之前计算的丑数,使用一个列表存储要计算的丑数。使用3个指针分别代表之前的丑数乘以2,3,5刚好大于当前列表最后一个丑数的位置。要添加的新的丑数即为3个指针所指的丑数分别乘以2,3,5之后的最小值,然后再更新2,3,5的指针位置。

猜你喜欢

转载自www.cnblogs.com/kingshine007/p/11494828.html