81 lines
2.1 KiB
Java
81 lines
2.1 KiB
Java
//超级丑数 是一个正整数，并满足其所有质因数都出现在质数数组 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 中的所有值都 互不相同 ，且按 递增顺序 排列
|
||
//
|
||
//
|
||
//
|
||
//
|
||
// Related Topics 数组 哈希表 数学 动态规划 堆（优先队列）
|
||
// 👍 249 👎 0
|
||
|
||
package leetcode.editor.cn;
|
||
|
||
import java.util.HashSet;
|
||
import java.util.PriorityQueue;
|
||
import java.util.Set;
|
||
|
||
//313:超级丑数
|
||
class SuperUglyNumber {
|
||
public static void main(String[] args) {
|
||
//测试代码
|
||
Solution solution = new SuperUglyNumber().new Solution();
|
||
}
|
||
|
||
//力扣代码
|
||
//leetcode submit region begin(Prohibit modification and deletion)
|
||
class Solution {
|
||
public int nthSuperUglyNumber(int n, int[] primes) {
|
||
PriorityQueue<Integer> pq = new PriorityQueue<>();
|
||
pq.add(1);
|
||
Set<Integer> seen = new HashSet<>();
|
||
for(int i=1;i<n;i++){
|
||
int cur = pq.poll();
|
||
for(int p: primes){
|
||
// 防止爆int处理
|
||
if(p > Integer.MAX_VALUE / cur)
|
||
break;
|
||
if(!seen.contains(cur * p)){
|
||
seen.add(cur * p);
|
||
pq.add(cur * p);
|
||
}
|
||
}
|
||
}
|
||
return pq.poll();
|
||
}
|
||
}
|
||
//leetcode submit region end(Prohibit modification and deletion)
|
||
|
||
} |