给你一个整数 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/ugly-number-ii
class Solution:
def nthUglyNumber(self, n: int) -> int:
dp = [0]*n
dp[0] = 1
a= b = c = 0
for i in range(1,n):
dp[i] = min(dp[a]*2,dp[b]*3,dp[c]*5)
if dp[i] == dp[a]*2:a+=1
if dp[i] == dp[b]*3:b+=1
if dp[i] == dp[c]*5:c+=1
return dp[n-1]
之前错误的思路
- 丑数一定是a个2,b个3, c个5 相乘,因此构建树,构建到第n层结束
- 想用深度优先搜索,然后剪枝去重
- 层次遍历这棵树,将节点加入二叉排序树中
- 最后中序遍历二叉排序树,得到第n个丑数
这个方法时间复杂度太高,n等于10的时候,第10个丑数是12,构建树的时候却要构建到1024,效率奇低