一、题目
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
二、思路
显然通过生成一个数组然后去判断是不是丑数这样的操作是很慢的这样会导致要判断很多没有必要的数,例如1, 2, 3, 4, 5, 6, 8, 9,10, 12, 15, 16, 都是丑数,但是7,11,13,14也会纳入判断的范围,时间复杂度会大大地提高。所以换一个思路就是既然丑数是通过2,3,5组成的那么也是可以通过2,3,5相乘拼接出一个新的数字。
三、代码实现
在range()的参数值为index-1的意思是指初始的ugly_list就有1这一个元素,所以当你求index = 1的时候得到结果是2,但是题目说的是第一个丑数是1。所以这里通过减一去达到第一个数为1这一个条件。其次index2,index3,index5是指有多少个2多少个3多少个5拼接而成,这里取它的最小值是想满足题目写的从小到大的条件。用这种计数法的好处还可以避免重复,大大提高效率。
四、代码实现
class Solution:
def GetUglyNumber_Solution(self, index):
# write code here
if index <= 0:
return 0
index2 = 0
index3 = 0
index5 = 0
ugly_list = [1]
for i in range(index):
temp = min(ugly_list[index2]*2, ugly_list[index3]*3, ugly_list[index5]*5)
ugly_list.append(temp)
if temp % 2 == 0:
index2 += 1
if temp %3 == 0:
index3 += 1
if temp % 5 == 0:
index5 += 1
return ugly_list[-1]
五、测试
if __name__ == '__main__':
so = Solution()
print(so.GetUglyNumber_Solution(13))
六、结果
20
Process finished with exit code 0
七、反思
凡是遇到类似于这一种题目,某种特殊的数可以被另外的的数整除,可以反过来想怎么去生成着一些数字,同时可以通过count有多少个这样的数相乘去避免重复,最后就是边界值和初始值的处理。