【LeetCode】超级丑数Java题解

238 阅读2分钟

这是我参与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;
    }
}

image.png

总结

  • 上述代码的时间复杂度是 O(log (n * m)), 空间复杂度是 O(n * m)
  • 坚持每日一题,加油!