标签:丑数 set Java int next II heap PriorityQueue
问题描述
题干:
给你一个整数 n ,请你找出并返回第 n 个 丑数 。
丑数 就是只包含质因数 2、3 或 5 的正整数。
示例1:
输入:n = 10
输出:12
解释:[1, 2, 3, 4, 5, 6, 8, 9, 10, 12] 是由前 10 个丑数组成的序列。
示例2:
输入:n = 1
输出:1
解释:1 通常被视为丑数。
题解分析
输出第n个丑数,首先我们要知道丑数的定义,定义就是题干说的那样,我们一般把1当作第一个丑数,依次往后排
当然思路很清晰就是生成丑数数组,最后返回第n个即可,但是怎么生成丑数的顺序数组呢
这里采用了Queue的一个实现类PriorityQueue,也就是一个最小堆模型,这样保证每次得到最小的元素
我们发现丑数肯定是2,3或者5的倍数,所以我们只要按照这个规律插入元素然后用PriorityQueue返回第n个即可
正确代码
public int nthUglyNumber(int n){
int[] factors = {2,3,5};
PriorityQueue<Long> heap = new PriorityQueue<>();
Set<Long> set = new HashSet();
set.add(1L);
heap.offer(1L);
int ugly = 0;
for (int i = 0; i < n; i++) {
long curr = heap.poll();
ugly = (int)curr;
//添加2、3或者5的倍数
for (int factor : factors) {
long next = curr * factor;
//排除重复元素
if (set.add(next)){
heap.offer(next);
}
}
}
return ugly;
}
总结
这里使用了一个PriorityQueue实现类,这个类是模仿了最小堆的结构,保证了每次返回的元素都是堆中的最小值
如果用数组保存的话虽然可以查看其它元素,但是当然也会添加额外的空间消耗,当然官网也有动态规划方法,效率也是比这个高的
当然每次这种题目,总会有大佬采用面向结果编程,直接把答案数组返回,这肯定是不能超越的速度
如果文章存在问题或者有更好的题解,欢迎在评论区斧正和评论,各自努力,你我最高处见
标签:丑数,set,Java,int,next,II,heap,PriorityQueue 来源: https://www.cnblogs.com/bc-song/p/14643642.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。