本题和之前的丑数不同,给出了不同质数的primes数组。
本题解题思路为新建一个tag数组,ugly表示从小到大的丑数序列,tag表示此质数将和第几个丑数相乘,因为之前的丑数已经乘过并且写入了丑数序列中。
先依次将数相乘得出第i个丑数的值,再将小于此丑数的tag下标加一,最后返回ugly【n-1】就是第n个丑数。
代码如下:
class Solution {
public int nthSuperUglyNumber(int n, int[] primes) {
int[] tag =new int[primes.length];
int[] ugly = new int[n];
ugly[0] = 1;
for (int i = 1; i < ugly.length; i++) {
ugly[i] = Integer.MAX_VALUE;
for (int j = 0; j < primes.length; j++) {
ugly[i] = Math.min(ugly[i], primes[j]*ugly[tag[j]]);
}
for (int j = 0; j < primes.length; j++) {
if(ugly[i]>=primes[j]*ugly[tag[j]]){
tag[j]++;
}
}
}
return ugly[n-1];
}
}