题目:我们把只包含因子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的指针位置。