LC313.超級醜數（這個數超級醜） - 動態規劃

題目描述

解題思路
這題的解法和第K個醜數（戳這裏）解法一樣，只是在第K個醜數的基礎上擴展了一下，下面的代碼就是對比第K個醜數寫出來的。

    int nthSuperUglyNumber(int n, vector<int>& primes) {
        // int idx2 = 0, idx3 = 0, idx5 = 0;
        vector<int> id(primes.size(), 0);
        vector<int> dp(n+1, 1);
        for(int i = 1; i <= n; i++)
        {
            // dp[i] = min(min(dp[idx2] * 2, dp[idx3] * 3), dp[idx5] * 5);
            dp[i] = dp[id[0]] * primes[0];
            for(int f = 1; f < id.size(); f++)
            {
                dp[i] = min(dp[id[f]] * primes[f],dp[i]);
            }
            // dp[i] == 2 * dp[idx2] ? idx2++ : idx2;
            // dp[i] == 3 * dp[idx3] ? idx3++ : idx3;
            // dp[i] == 5 * dp[idx5] ? idx5++ : idx5;
            for(int f = 0; f < id.size(); f++)
            {
                dp[i] == dp[id[f]] * primes[f] ? id[f]++ : id[f];
            }
        }
        return dp[n-1];
    }