剑指offer-python：45.丑数

题目：把只包含质因子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]