题目:
剑指 Offer 49. 丑数。
我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
链接:https://leetcode-cn.com/leetbook/read/illustrate-lcof/50qoxc/
本题与主站 264 题相同:https://leetcode-cn.com/problems/ugly-number-ii/
示例
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:
1 是丑数。
n 不超过1690。
思路:
除第一个丑数外,其余丑数皆为2或3或5的倍数。
每次取2,3,5倍数的最小值存入列表中,倍数通过已存入列表的值控制。
最后将第n个数丑数返回。
代码:
class Solution:
def nthUglyNumber(self, n: int) -> int:
arr = [1]
a, b, c = 0, 0, 0
for i in range(1, n):
times2, times3, times5= arr[a]*2, arr[b]*3, arr[c]*5
arr.append(min(min(times2, times3), times5))
if arr[-1] == times2:
a += 1
if arr[-1] == times3:
b += 1
if arr[-1] == times5:
c += 1
return arr[n-1]
时间复杂度O(n),空间复杂度O(n).