剑指Offer49-丑数

剑指Offer49-丑数

我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。

示例:


输入: n = 10

输出: 12

解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。

说明:

  • 1 是丑数。

  • n 不超过1690。

来源:力扣(LeetCode)

链接:力扣

一解

根据定义,丑数为 2^x * 3^y * 5^z ,其中 x, y, z 属于 n(0, 1, 2,…),只要根据上述公式依次进行计算即可。

设数组 a 存储丑数,从第一轮计算中最小值 p1, p2, p3 都是 1 ,加入 a :


class Solution:

    def nthUglyNumber(self, n: int) -> int:

        p2, p3, p5 = 0, 0, 0

        a = [1]

        for i in range(2, n+1):

            new_num = min(a[p2] * 2, a[p3] * 3, a[p5] * 5)

            a.append(new_num)

            if new_num == a[p2] * 2: p2 += 1

            if new_num == a[p3] * 3: p3 += 1

            if new_num == a[p5] * 5: p5 += 1

        return a[-1]

有没有漂亮数