考虑动态规划法,因为要找出第n个丑数,所以我们可以从第一个开始依次生成丑数,直到生成第n个即可。但是一定要保证丑数序列是单调递增的。丑数都是由2,3,5组成,所以如果有一个数是丑数,那么乘以2或乘以3或乘以5仍然是丑数。所以依据这个特性,记录p2,p3,p5三个指针用来指示打算要乘以的那个丑数,然后计算这三个指针乘以对应的丑数的最小值,就是当前位置的丑数,记得要更新p2,p3,p5,如果p2,p3,p5已经对这个丑数做过乘法操作,那就不能做重复操作,否则产生重复丑数。
class Solution(object):
def nthUglyNumber(self, n):
"""
:type n: int
:rtype: int
"""
dp = []
dp.append(1)
p2 = 0
p3 = 0
p5 = 0
for i in range(1, n):
temp = min(dp[p2] * 2, dp[p3] * 3, dp[p5] * 5)
if temp == (dp[p2] * 2):
p2 += 1
if temp == (dp[p3] * 3):
p3 += 1
if temp == (dp[p5] * 5):
p5 += 1
dp.append(temp)
return dp[-1]