PROBLEM

剑指 Offer 49. 丑数

难度 中等

MY ANSWER

逐个数检验，TLE。

BETTER SOLUTION

动态规划，根据前面的数生成下一个丑数。时间复杂度O(n)，空间复杂度O(n)。

Picture1.png

image-20210920020212841

class Solution {
public:
    int nthUglyNumber(int n) {
        int a = 0, b = 0, c = 0;
        int dp[n];
        dp[0] = 1;
        for(int i = 1; i < n; i++) {
            int n2 = dp[a] * 2, n3 = dp[b] * 3, n5 = dp[c] * 5;
            dp[i] = min(min(n2, n3), n5);
            if(dp[i] == n2) a++;
            if(dp[i] == n3) b++;
            if(dp[i] == n5) c++;
        }
        return dp[n - 1];
    }
};

SUMMARY

注意怎样使用动态规划来生成数。