面试题49. 丑数

题目

我们把只包含因子 2、3 和 5 的数称作丑数（Ugly Number）。求按从小到大的顺序的第 n 个丑数。

 

示例:

输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:  

1 是丑数。
n 不超过1690。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/chou-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

除第一个丑数以外，每一个丑数都等于之前的某一个数乘上2,3,或者5中某一个数得到的最小的那一个数，也就是min（a*2，b*3，c*5），但是这些数需要满足条件是a*2>x>a-1*2，b*2>x>b-1*2，c*2>x>c-1*2。

第一步定义数组dp[1]，初始化abc；第二步进入循环判断min；第三步每进行一次判断之后，得到一个丑数，判断该丑数是min中的哪一个，对相应的abc索引进行加一；最后输出dp[-1]

代码

class Solution:
    def nthUglyNumber(self, n: int) -> int:
        dp=[1]*n
        a,b,c=0,0,0
        for i in range(1,n):
            n2,n3,n5=dp[a]*2,dp[b]*3,dp[c]*5
            dp[i]=min(n2,n3,n5)
            if dp[i]==n2:a+=1
            if dp[i]==n3:b+=1
            if dp[i]==n5:c+=1
        return dp[-1]