【LeetCode】超级丑数 Java 题解

关注

发布于: 6 小时前

题目描述

超级丑数是一个正整数，并满足其所有质因数都出现在质数数组 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<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)
坚持每日一题，加油！

发布于: 6 小时前阅读数: 4

原文链接:【http://xie.infoq.cn/article/4b21204060806cc46aadb15fd】。文章转载请联系作者。

HQ数字卡

关注

还未添加个人签名 2019.09.29 加入

LeetCode,略懂后端的RD

发布

暂无评论

创作场景

【LeetCode】超级丑数 Java 题解

题目描述

思路分析

代码

总结

HQ数字卡

评论