题目
编写一段程序来查找第 n 个超级丑数。
超级丑数是指其所有质因数都是长度为 k 的质数列表 primes 中的正整数。
示例:
输入: n = 12, primes = [2,7,13,19]
输出: 32
解释: 给定长度为 4 的质数列表 primes = [2,7,13,19],前 12 个超级丑数序列为:[1,2,4,7,8,13,14,16,19,26,28,32] 。
说明:1 是任何给定 primes 的超级丑数。 给定 primes 中的数字以升序排列。 0 < k ≤ 100, 0 < n ≤ 106,
0 < primes[i] < 1000 。 第 n 个超级丑数确保在 32 位有符整数范围内。来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/super-ugly-number
解题思路
第一步:根据超级丑数都是质数相乘而来,定义一个结果数组res,res[0]=1;
第二步:为每个质数记录一个下标,该小标表示:乘对应质数能得到的最小值的下标;
第三步:比较所有质数所能得到最小值,得出最小的一个,取出;一直查找直到找到第n个。
代码如下
class Solution {
public int nthSuperUglyNumber(int n, int[] primes) {
int[] res = new int[n];
//存储对应质数所到达的下标
int[] indexs = new int[primes.length];
res[0] = 1;
for (int i = 1; i < n; i++) {
int min = Integer.MAX_VALUE;
for (int j = 0; j < primes.length; j++) {
if (primes[j] * res[indexs[j]] < min) {
min = primes[j] * res[indexs[j]];
}
}
res[i] = min;
//因为得到最小的可能有多个值,所以把全部可达最小值的质数对应下标加一
for (int j = 0; j < primes.length; j++) {
if (primes[j] * res[indexs[j]] == min) {
indexs[j]++;
}
}
}
return res[n-1];
}
}