写一个程序来找第 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 作为结果。
思路:
每个丑数都是有primes中的元素与另一个丑数相乘得到的,可以从1开始,依次与primes中元素相乘,
数字存入set,取最小值最为新的丑数,在将新丑数与primes中元素相乘,存入set,得到最小值就是第二个新的丑数。
#ifndef C518_H
#define C518_H
#include<iostream>
#include<vector>
#include<set>
using namespace std;
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
if (primes.empty())
return 0;
set<long long> se;
long long num = 1;
while (n > 1)
{
for (auto c : primes)
se.insert(c*num);
num = *se.begin();
se.erase(se.begin());
n--;
}
return num;
}
};
#endif