给你一个整数 n ,请你找出并返回第 n 个 丑数 。
丑数 就是只包含质因数 2、3 和/或 5 的正整数。
示例 1:输入:n = 10
输出:12
解释:[1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 个丑数组成的序列。
示例 2:输入:n = 1
输出:1
解释:1 通常被视为丑数。提示:
1 <= n <= 1690
思路分析:
这道题可以用dp来做,首先创建一个初始的数组,数组起始元素均为1,然后定义三个元素用来2、3、5的计数,遍历dp中的所有元素,每个元素都取对应元素个数乘完2、3、5后的最小值,直到给第十个元素赋值即为第十个丑数。
class Solution:
def nthUglyNumber(self, n: int) -> int:
if n < 0:
return
index2 = 0
index3 = 0
index5 = 0
dp = [1 for i in range(n)]
for i in range(1, n):
dp[i] = min(dp[index2] * 2, dp[index3] * 3, dp[index5] * 5)
if dp[i] == dp[index2] * 2:
index2 += 1
if dp[i] == dp[index3] * 3:
index3 += 1
if dp[i] == dp[index5] * 5:
index5 += 1
return dp[n-1]