浅谈 丑数 II 问题

丑数 II

问题：
给你一个整数 n ，请你找出并返回第 n 个 丑数 。

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

思路：
动态规划。
意图将从 1 ~ n 的丑数都求出来，然后返回最后一个丑数。
创建一个存放过程中丑数的数组。
分别计算下一个丑数，将最小的放入数组，得出的数字指针加一。
持续重复上一步骤，直到算出第 n 个丑数。

class Solution {
public:
    int nthUglyNumber(int const n) const {
        if(n > 0 && n < 7) return n;		// 1 ~ 6 为丑数
        vector<int> dp(n + 1, 0);
        dp[1] = 1;
        int p2 = 1, p3 = 1, p5 = 1;
        for(auto i = 2; i <= n; ++i) {
            int num1= dp[p2] * 2, num2 = dp[p3] * 3, num3 = dp[p5] * 5;
            dp[i] = min(min(num1, num2), num3);
            if(dp[i] == num1) {
                p2++;
            }
            if(dp[i] == num2) {
                p3++;
            }
            if(dp[i] == num3) {
                p5++;
            }
        }
        return dp.back();
    }
};

• 时间复杂度：O(n)
• 空间复杂度：O(n)