编写一段程序来寻找第 n 个超级丑数。
超级丑数是指其所有质因数都在长度为k的质数列表primes中的正整数。例如,[1, 2, 4, 7, 8, 13, 14, 16, 19, 26, 28, 32],是给定长度为 4 的质数列表primes = [2, 7, 13, 19]的前 12 个超级丑数。
注意:
(1) 1是任何给定的primes的超级丑数。
(2) 给定primes中的数字以升序排列。
(3) 0 < k ≤ 100, 0 < n ≤ 106, 0 < primes[i] < 1000 。
(4) 第n个超级丑数可以认为在32位有符整数的表示范围内。
详见:https://leetcode.com/problems/super-ugly-number/description/
C++:
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();
}
};
参考:https://www.cnblogs.com/grandyang/p/5144918.html