写一个程序来找第 n 个超级丑数。
超级丑数的定义是正整数并且所有的质数因子都在所给定的一个大小为 k 的质数集合内。
比如给你 4 个质数的集合 [2, 7, 13, 19]
, 那么 [1, 2, 4, 7, 8, 13, 14, 16, 19, 26, 28, 32]
是前 12 个超级丑数。
注意事项
1
永远都是超级丑数不管给的质数集合是什么。- 给你的质数集合已经按照升序排列。
- 0 < k ≤ 100, 0 < n ≤ 10^6, 0 < primes[i] < 1000
给出 n = 6
和质数集合 [2, 7, 13, 19]
。第 6 个超级丑数为 13,所以返回 13
作为结果
class Solution {
public:
/**
* @param n a positive integer
* @param primes the given prime list
* @return the nth super ugly number
*/
int nthSuperUglyNumber(int n, vector<int>& primes) {
// Write your code here
int num = primes.size();
vector<int> superugly(n, 0);
superugly[0] = 1;
vector<int> idxprimer(num, 0);
int count = 1;
while(count < n)
{
int min = 0x7fffffff;
for(int i=0; i<num; ++i)
{
int temp = superugly[idxprimer[i]] * primes[i];
min = temp > min ? min : temp;
}
for(int i=0; i<num; ++i)
{
if(min == superugly[idxprimer[i]] * primes[i])
{
++idxprimer[i];
superugly[count] = min;
}
}
++count;
}
return superugly[n-1];
}
};
这题的关键在于使用一个数组idxprimer来记录中间量。