题目:
我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
leetcode原题链接
思路:
参考题解
原始思路:判断每个数是不是丑数,直到丑数数量达到n,该方法会超时
利用丑数的递推性质丑数只包含因子 2, 3, 5 ,因此有 “丑数 == 某较小丑数 × 2/3/5”
代码:
class Solution:
def nthUglyNumber(self, n: int) -> int:
dp=[1]*n#创建长度为n的动态规划列表,每个元素初始值为1,这样就不用单独设定dp[0]=1
a,b,c=0,0,0
i=1#从下标为1的元素开始
while i<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
i+=1
return dp[n-1]#返回第n个丑数
新发现:
创建空字典 比 创建已知长度的数组 耗时
第三行代码分别为dp=[1]*n
(左)和dp={}
【需要额外加代码dp[0]=1
】(右)的运行效果