题目的意思就是根据给定的素数数组为因子,求第n个丑数。解法其实和ugly number II 差不多,只不过输入的素数个数是不确定的。最简单的方法就是枚举,直到找到第N个丑数,但是因为你不知道输入的因子,所以你枚举的数量就需要非常大,效率也就会变得非常糟糕。
还是利用之前的方法,重复的利用已经计算出来的丑数来确定下一个丑数,直到找到第N个。首先,我们需要确定到底有几个素数因子,下一个丑数一定是之前的某个丑数乘以某个素数因子得到的,所以我们可以枚举出所以当前丑数分别乘以素数因子得到的新丑数,然后选取刚好大于当前最大丑数的数为下一丑数。这里还是同样的问题,随着丑数的数量的增多,计算得到的新丑数的量也不断增加,有没有啥方法能够减少计算量。我们可以为每个素数因子维护一个最小索引值。这个最小索引值代表的意思就是,该素数因子乘以对应索引为的丑数刚好能够大于当前所得到的最大丑数。这样,每一次我们只需要从这些丑数中选取最小的,就是下一个丑数。
代码如下:
class Solution {
public:
long long nthSuperUglyNumber(int n, vector<int>& primes) {
if(n==1)
return 1;
int size = primes.size();
vector<int> index(size,0);
vector<long long> ugly;
ugly.push_back(1);
//每次都需要记录所有因子对应的最小索引(这个数乘以对应因子一定大于当前获得的ugly number,每次选取最小的)
int cur = 0;
while(cur < n){
vector<long long> temp(size);
for(int i=0;i<size;++i){
temp[i]=ugly[index[i]]*primes[i];
}
auto m = min_element(temp.begin(),temp.end());
//取出最小的数并加入到ugly中
ugly.push_back(*m);
//更新index中的最小索引
for(int i=0;i<size;++i){
while(ugly[index[i]]*primes[i]<=*m){
++index[i];
}
}
++cur;
}
return ugly[n-1];
}
};
public:
long long nthSuperUglyNumber(int n, vector<int>& primes) {
if(n==1)
return 1;
int size = primes.size();
vector<int> index(size,0);
vector<long long> ugly;
ugly.push_back(1);
//每次都需要记录所有因子对应的最小索引(这个数乘以对应因子一定大于当前获得的ugly number,每次选取最小的)
int cur = 0;
while(cur < n){
vector<long long> temp(size);
for(int i=0;i<size;++i){
temp[i]=ugly[index[i]]*primes[i];
}
auto m = min_element(temp.begin(),temp.end());
//取出最小的数并加入到ugly中
ugly.push_back(*m);
//更新index中的最小索引
for(int i=0;i<size;++i){
while(ugly[index[i]]*primes[i]<=*m){
++index[i];
}
}
++cur;
}
return ugly[n-1];
}
};