题目:

写一个程序来找第 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数组需要好好想想。
clipboard.png

参考答案:

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];
    }
};

wydong
40 声望5 粉丝

wyd


下一篇 »
实现 Trie