34、丑数

2018-10-04  本文已影响0人  小碧小琳

题目描述
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。

一开始最直观的的想法就是写出一个函数,输入数字能够判断是不是丑数。然后一直遍历所有的数字,一直到第N个丑数出现为止。
但是太耗时间。

解决办法:用空间换时间

因为太耗时间,于是我们想到用空间换时间的方式。我们发现,丑数可以由另外的丑数计算而来。比如,8由4乘以2得到。
因此

还能改进:

那么每次为了得到一个新的丑数,都需要用已有的丑数序列乘以2,3,5然后去重,筛选出最小的值吗?显然有很多是重复计算了的。

剑指offer的方法中指出,我们只需要找出特定的三个数字,然后用这三个数字分别乘以2,3,5得到三个数,选择其中最小值即可。

————————————————————————————————————
那么怎么确定这三个数呢?
因为上面已经说了,如果用所有的已有丑数乘以2,3,5会造成重复,那么从哪里开始,就不重复了呢?
比如上述的丑数序列中,乘以2为例,我们可以发现:

————————————————————————————————————

由上解释,可以知道改进方法,在乘以2时,用丑数中每个数乘以2,当得到的值刚大于序列中最大值M时,就把该值记作M2。同理,丑数序列中每个数乘以3,乘以5,得到M3,M5。然后选择其中最小的数作为新的丑数。就这样一直到第N个丑数。

综上,就是用一个长度为N的数列来存储N个丑数,达到以空间换时间的目的。

代码实现:

# -*- coding:utf-8 -*-
class Solution:
    def GetUglyNumber_Solution(self, index):
        if index == 0:
            return 0
        if index == 1:
            return 1

        #初始化丑数序列
        res = [1]
        for i in range(1,index):
            #根据已有丑数序列,得到新的三个丑数候选值
            M2_index = 0
            M3_index = 0
            M5_index = 0
            while res[M2_index]*2 <= res[-1]:
                M2_index += 1
            M2 = res[M2_index]*2

            while res[M3_index]*3 <= res[-1]:
                M3_index += 1
            M3 = res[M3_index]*3

            while res[M5_index]*5 <= res[-1]:
                M5_index += 1
            M5 = res[M5_index]*5

            #添加新的丑数
            res.append(min(M2,M3,M5))

        return res[-1]

S = Solution()
S.GetUglyNumber_Solution(6)
上一篇下一篇

猜你喜欢

热点阅读