0
点赞
收藏
分享

微信扫一扫

剑指offer-python:45.丑数

小暴龙要抱抱 2022-02-11 阅读 50
python

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

解析:分别保存*2、*3、*5的最小位置,每次与该位置的数*2、3、5分别对比,即可。

代码:

class Solution:
    def func(self , n):
        if n == 0:
            return 0
        res = []
        n2 , n3 , n5 = 0,0,0
        for i in range(n):
            if i == 0:
                res.append(1)
            else:
                r2 , r3 , r5 = res[n2]*2 , res[n3]*3 , res[n5]*5
                to_add = min(r2 , r3 , r5)
                res.append(to_add)
                if r2 == to_add:
                    n2 += 1
                if r3 == to_add:
                    n3 += 1
                if r5 == to_add:
                    n5 += 1
        return res
a = 10
s = Solution()
print(s.func(a))

输出:

[1, 2, 3, 4, 5, 6, 8, 9, 10, 12]
举报

相关推荐

0 条评论