题目描述
解题思路
这题的解法和第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];
}