题目:
写一个程序来找第 n 个超级丑数。
超级丑数的定义是正整数并且所有的质数因子都在所给定的一个大小为 k 的质数集合内。
比如给你 4 个质数的集合 [2, 7, 13, 19], 那么 [1, 2, 4, 7, 8, 13, 14, 16, 19, 26, 28, 32] 是前 12 个超级丑数。
注意事项:
- 永远都是超级丑数不管给的质数集合是什么。
- 给你的质数集合已经按照升序排列。
- 0 < k ≤ 100, 0 < n ≤ 10^6, 0 < primes[i] < 1000
样例:
给出 n = 6 和质数集合 [2, 7, 13, 19]。第 6 个超级丑数为 13,所以返回 13 作为结果。
思路:
丑数产生的规律:除了第一个数字,其余所有数字都是之前已有数字乘以任意一个在质数数组里的质数得到。
point数组需要好好想想。
参考答案:
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 len = primes.size();
vector<int> point(len,0);//用于保存当前的位置,这个需要仔细想想
vector<int> ugly(n,INT32_MAX);//不要用INT16_MAX,有些测试用例过不了
ugly[0] = 1;
for(int i=1; i<n; i++){
for(int j=0; j<len; j++){
ugly[i] = min(ugly[i],ugly[point[j]]*primes[j]);
}
for(int j=0; j<len; j++){
if(ugly[i] == ugly[point[j]]*primes[j]){
point[j]++;
}
}
}
return ugly[n-1];
}
};
**粗体** _斜体_ [链接](http://example.com) `代码` - 列表 > 引用
。你还可以使用@
来通知其他用户。