剑指 Offer 49. 丑数(动态规划)

剑指 Offer 49. 丑数

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

 

示例:

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

说明:  

1. 1 是丑数。
2. n 不超过1690。

class Solution {
public:
    int nthUglyNumber(int n) {
        // 这里当给定值非法时,返回-1
        if (n <= 0) {
            return -1;
        }
        vector<int> dp(n, 0);
        dp[0] = 1;
        int a = 0;
        int b = 0;
        int c = 0;
        for (int i = 1; i < n; i++) {
            int n2 = dp[a] * 2;
            int n3 = dp[b] * 3;
            int n5 = dp[c] * 5;
            dp[i] = min(min(n2, n3), n5);
            if (n2 == dp[i]) {
                a++;
            }
            if (n3 == dp[i]) {
                b++;
            }
            if (n5 == dp[i]) {
                c++;
            }
        }
        return dp[n - 1];
    }
};