超级丑数 是一个正整数,并满足其所有质因数都出现在质数数组 primes 中。
给你一个整数 n 和一个整数数组 primes ,返回第 n 个 超级丑数 。
题目数据保证第 n 个 超级丑数 在 32-bit 带符号整数范围内。
示例 1:
输入: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] 。
示例 2:
输入:n = 1, primes = [2,3,5]
输出:1
解释:1 不含质因数,因此它的所有质因数都在质数数组 primes = [2,3,5] 中。
提示:
1 <= n <= 106
1 <= primes.length <= 100
2 <= primes[i] <= 1000
题目数据 保证 primes[i] 是一个质数
primes 中的所有值都 互不相同 ,且按 递增顺序 排列
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/super-ugly-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
和丑数的解法类似, 只是丑数是三指针,而此题变成多指针了,指针的个数就是 质数数组 primes 的长度,每次求出当前指针乘积的最小值,放入 ans 数组中,然后找到乘积为该数字的指针,让它们加加即可,具体思路见代码:
class Solution {
public int nthSuperUglyNumber(int n, int[] primes) {
int[] zhen = new int[primes.length];// 指针数组
int[] ans = new int[n + 1];
ans[0] = 1;
for (int i = 1; i < n; i++) {
int next = Integer.MAX_VALUE;// next 为下一个位置的值
for (int j = 0; j < primes.length; j++) {
int temp = ans[zhen[j]] * primes[j];// 求出当前指针乘积
if (temp < next && temp > ans[i - 1]) {
next = temp;// 找到最小的 next,并且比前一个大的
}
}
ans[i] = next;// 赋值给 ans 数组,然后比较关键
// 这一部的循环目的是找到指针乘积等于 next 的指针,但是有人可能会觉得;
// 那我直接定义一个指针,在上面直接更新不就行了,但这样想不行
// 因为可能会有重复的结果,比如,指针 2 和 指针 7,都可能得到 14 的结果。
for (int j = 0; j < primes.length; j++) {
if (ans[zhen[j]] * primes[j] == next) {
zhen[j]++;
}
}
}
return ans[n - 1];
}
}