编写一段程序来寻找第 n 个超级丑数。
超级丑数是指其所有质因数都在长度为k的质数列表primes中的正整数。例如,[1, 2, 4, 7, 8, 13, 14, 16, 19, 26, 28, 32],是给定长度为 4 的质数列表primes = [2, 7, 13, 19]的前 12 个超级丑数。
注意:
(1) 1是任何给定的primes的超级丑数。
(2) 给定primes中的数字以升序排列。
(3) 0 < k ≤ 100, 0 < n ≤ 106, 0 < primes[i] < 1000 。
(4) 第n个超级丑数可以认为在32位有符整数的表示范围内。
详见:https://leetcode.com/problems/super-ugly-number/description/
C++:
class Solution { public: int nthSuperUglyNumber(int n, vector<int>& primes) { vector<int> res(1,1),idx(primes.size(),0); while(res.size()<n) { vector<int> tmp; int mn=INT_MAX; for(int i=0;i<primes.size();++i) { tmp.push_back(res[idx[i]]*primes[i]); } for(int i=0;i<primes.size();++i) { mn=min(mn,tmp[i]); } for(int i=0;i<primes.size();++i) { if(mn==tmp[i]) { ++idx[i]; } } res.push_back(mn); } return res.back(); } };
参考:https://www.cnblogs.com/grandyang/p/5144918.html