313 超级丑数
题目分析:这道题目与264 丑数II几乎一模一样,只是丑数的质因数不再是固定的2、3、5,而是通过数组给出。不过不影响解题。
思路:开辟一个primes.size()大小的数组,每个元素初始化为0。其余步骤与 264 丑数II 一样。唯一需要注意的是,可能下一个丑数有不止一种产生方式 UglyNum[index[i]]*primes[i]==UglyNum[index[j]]*primes[j],比如12=4*3=6*2,那么应该将两个质因数对应的标签都加1;需要理解的部分在代码中给出注释。
int nthSuperUglyNumber(int n, vector<int>& primes) {
vector<int> UglyNum(n,0);
int bucket=primes.size();
vector<int> index(bucket,0);
int i=1;
UglyNum[0]=1;
int Min=INT_MAX;
while(i<n){
Min=INT_MAX;
for(int k=0;k<bucket;k++)
Min=min(Min,UglyNum[index[k]]*primes[k]); //从bucket个丑数中找出最小的
UglyNum[i]=Min;
i++;
for(int k=0;k<bucket;k++){ //找到最小丑数是由哪些质因数和其对应的已存在丑数产生,并将该质因数的标签+1
if(Min==UglyNum[index[k]]*primes[k]){ //如果Min==质因数乘以对应位置的丑数,有不止一种产生方式,则每种对应的质因数标签都+1
index[k]++;
}
}
}
return UglyNum[n-1];
}