参考代码:
class Solution {
public int nthSuperUglyNumber(int n, int[] primes) {
int[] dp=new int[n];
dp[0]=1;
int[] index=new int[primes.length];
for(int i=1;i<n;i++){
int min=Integer.MAX_VALUE;
for(int j=0;j<primes.length;j++){
min=Math.min(primes[j]*dp[index[j]],min);
}
for(int j=0;j<primes.length;j++){
if(primes[j]*dp[index[j]]==min) index[j]++;
}
dp[i]=min;
}
return dp[n-1];
}
}
大致思路:这题与丑数二非常相似,丑数二只是将数组中的数固定为2,3,5,但是这题可以是任意取值,所以我们需要加入循环,让数组中的每一个数都遍历到,找到最小值,再将对应的值加一,最后返回最后一个数。
具体的思路我就不赘述了,和丑数二的思路一样,就是家里两个循环,将特例变成一般。