这是我参与8月更文挑战的第8天,活动详情查看:8月更文挑战
题目描述
超级丑数 是一个正整数,并满足其所有质因数都出现在质数数组 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] 。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/super-ugly-number
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路分析
- 这是一个常见题目,一句话理解题目,将从1到n, 满足丑数条件的元素排序,输出第 n 个丑数。
- 丑数的条件,题目已经给出,满足其所有质因数都出现在质数数组 primes 中。
- 接下来,还需要排序功能。一般使用 Heap 数据结构解决。在实际使用中,一般使用PriorityQueue。PriorityQueue是基于优先堆的一个无界队列,这个优先队列中的元素可以默认自然排序或者通过提供的Comparator(比较器)在队列实例化的时排序。要求使用Java Comparable和Comparator接口给对象排序,并且在排序时会按照优先级处理其中的元素。
- 除了PriorityQueue, Heap 还有多种实现方式,比如:二叉堆。
- 此题目还对数据范围进行了提示,在 32-bit 带符号整数范围内。因此,可以用 int 表示这个数据。
通过代码
class Solution {
public int nthSuperUglyNumber(int n, int[] primes) {
Set<Long> set = new HashSet<>();
set.add(1L);
PriorityQueue<Long> priorityQueue = new PriorityQueue<>();
priorityQueue.offer(1L);
int ugly = 0;
for (int i = 0; i < n; i++) {
long curr = priorityQueue.poll();
ugly = (int) curr;
for (int prime : primes) {
Long nextNum = curr * prime;
if (!set.contains(nextNum)) {
set.add(nextNum);
priorityQueue.offer(nextNum);
}
}
}
return ugly;
}
}
总结
- 上述代码的时间复杂度是 O(log (n * m)), 空间复杂度是 O(n * m)
- 坚持每日一题,加油!