剑指 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/
解题思路:
代码:
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]