LeetCode-264. 丑数 II

264. 丑数 II

编写一个程序，找出第 n 个丑数。

丑数就是只包含质因数 2, 3, 5 的正整数。

示例:

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

说明:

• 1 是丑数。
• n 不超过1690。

思路

首先想一想，丑数是由只含2，3，5的因子组成的，那一个丑数乘上2，3，5仍会得到一个丑数，所以生成新丑数可以以旧丑数为基础，2，3，5作为丑数的基；如何使生成的丑数保持递增的顺序，每次循环会生成一个新丑数，与原来两个旧丑数做比较，取最小的加入数组中。
首先初始化dp[0]=1,第一次循环，dp[0]*2,dp[0]*3,dp[0]*5,取最小2，更新l2=1，第二次循环，dp[1]*2,dp[0]*3,dp[0]*5,取最小3，更新l3=1,第三次循环，dp[1]*2,dp[1]*3,dp[0]*5,取最小4，更新l2=3,如此下去，得出结果

class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int> dp(n); //递增数组
        dp[0] = 1;
        int l2=0,l3=0,l5=0;
        for(int i=1;i<n;i++)
        {
            dp[i] = min(min(dp[l2]*2,dp[l3]*3),dp[l5]*5);//将最小的存入数组
            if(dp[i]==dp[l2]*2)
                l2++;
            if(dp[i]==dp[l3]*3)
                l3++;
            if(dp[i]==dp[l5]*5)
                l5++;
        }
        return dp[n-1];
    }
};