剑指 Offer 49. 丑数 | 算法

107 阅读1分钟

剑指 Offer 49. 丑数

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

示例:

输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:  

1 是丑数。
n 不超过1690。
注意:本题与主站 264 题相同:https://leetcode-cn.com/problems/ugly-number-ii/

解题思路

www.bilibili.com/video/BV1CK…

代码

class Solution:
    def nthUglyNumber(self, n: int) -> int:
        if n < 7: return n
        a = [1]
        p2 = p3 = p5 = 0
        for i in range(2, n+1):
            newNum = min(a[p2]*2, a[p3]*3, a[p5]*5)
            a.append(newNum)
            if newNum == a[p2]*2: p2 += 1
            if newNum == a[p3]*3: p3 += 1
            if newNum == a[p5]*5: p5 += 1
        return a[-1]