这道题给出一个质数数组,求出第n个超级丑陋数。
方法和之前的丑陋数一样,下面引用了别人博客里的图片,画的很清楚。这里需要一个idx数组,来记录primes里面每一个数应该和结果集里的哪一个数相乘。
class Solution {
public:
int nthSuperUglyNumber(int n, vector<int>& primes) {
vector<int> res(1, 1), idx(primes.size(), 0);
while(res.size() < n){
vector<int> tmp;
int mn = INT_MAX;
for(int i = 0; i < primes.size(); ++i){
tmp.push_back(res[idx[i]] * primes[i]);
}
for(int i = 0; i < primes.size(); ++i){
mn = min(mn, tmp[i]);
}
for(int i = 0; i < primes.size(); ++i){
if(mn == tmp[i])
++idx[i];
}
res.push_back(mn);
}
return res.back();
}
};