题目描述

超级丑数 是一个正整数，并满足其所有质因数都出现在质数数组 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是基于优先堆的一个无界队列，这个优先队列中的元素可以默认自然排序或者通过提供的Comparator（比较器）在队列实例化的时排序。要求使用Java Comparable和Comparator接口给对象排序，并且在排序时会按照优先级处理其中的元素。

• 此题目还对数据范围进行了提示，在 32-bit 带符号整数范围内。因此，可以用 int 表示这个数据。

代码

class Solution {
    public int nthSuperUglyNumber(int n, int[] primes) {
        Set set = new HashSet<>();
        set.add(1L);
        
        PriorityQueue 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)

• 坚持每日一题，加油！