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];
    }