題目:
寫一個程序來找第 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 hereint 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];}
};