题目描述

给你一个整数 n ,请你找出并返回第 n 个 丑数 。

  • 丑数 就是只包含质因数 2、3 和/或 5 的正整数。
  • 提示:1也是丑数

示例 1:

输入:n = 9
输出:10
解释:[1, 2, 3, 4, 5, 6, 8, 9, 10] 是由前 9 个丑数组成的序列。

解题思路

方案一,使用优先队列+哈希表:

  • 准备PriorityQueue<Long> priorityQueue用于结果排序; 和Set<Long> set 记录是否出现过数字;
  • 将1加入到priorityQueue和set中,并且新增count=1,记录添加的数字个数;
  • 从priorityQueue中取最小的数字v(priorityQueue.offer()),count++, 退出条件count>=n;然后执行v2=v*2;v3=v*3;v5=v*5; 
    • 如果set中不包含v2,则把v2加入到priorityQueue中,
    • 如果set中不包含v3,则把v3加入到priorityQueue中,
    • 如果set中不包含v5,则把v5加入到priorityQueue中,
  • 从priorityQueue中找到第n个数字;直接返回priorityQueue.offer();

代码实现如下:

import java.util.HashSet;
import java.util.PriorityQueue;
import java.util.Set;class Solution1 {public int nthUglyNumber(int n) {PriorityQueue<Long> priorityQueue = new PriorityQueue<>(n * 3);Set<Long> set = new HashSet<>(n * 3);if (n == 1) {return 1;}priorityQueue.offer(1L);set.add(1L);int count = 1;while (count < n) {long v = priorityQueue.poll();long v2 = 2 * v;if (!set.contains(v2)) {priorityQueue.offer(v2);set.add(v2);}long v3 = 3 * v;if (!set.contains(v3)) {priorityQueue.offer(v3);set.add(v3);}long v5 = 5 * v;if (!set.contains(v5)) {priorityQueue.offer(v5);set.add(v5);}count++;}return priorityQueue.poll().intValue();}public static void main(String[] args) {Solution1 solution = new Solution1();System.out.println(solution.nthUglyNumber(10));}
}

方案二,使用动态规划,核心思路是找到数学规律:

  • 准备p2,p3,p5作为2、3、5的父亲节点的下标,int dp[] = new int[n+1];
  • 初始化dp[1]=1; p2=1,p3=1,p5=1;
  • 从2到n做循环
    • 计算 v2=dp[p2] * 2 ; v3=dp[p3]*3 ; v5=dp[p5]*5;
    • 取v2、v3、v5中的最小值为min,并且dp[i]= min;
    • 判断: 如果min==v2 则p2++; 如果min==v3 则p3++; 如果min==v5 则p5++;
  • 直接返回dp[n];

代码实现如下:


class Solution {public int nthUglyNumber(int n) {// 使用dpint[] dp = new int[n + 1];dp[1] = 1;int p2 = 1;int p3 = 1;int p5 = 1;for (int i = 2; i <= n; i++) {int num2 = 2 * dp[p2];int num3 = 3 * dp[p3];int num5 = 5 * dp[p5];dp[i] = Math.min(Math.min(num2, num3), num5);if (num2 == dp[i]) {p2++;}// 坑点,如果是6,则p2和p3都要做自加操作if (num3 == dp[i]) {p3++;}if (num5 == dp[i]) {p5++;}}return dp[n];}

第二种方案耗时只有2ms,剩下的方案只能提前算出来数字存储到array里面能更快了吧。

题目扩展-超级丑数

超级丑数 是一个正整数,并满足其所有质因数都出现在质数数组 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] 。

第二种方案,对代码进行改造,将确定的数字,切换成数组;核心点就是对原来的2,3,5改造成数组计算,这样通用性更高,代码实现如下:

import java.util.Arrays;class Solution {public int nthSuperUglyNumber(int n, int[] primes) {long[] dp = new long[n + 1];dp[1] = 1;int p[] = new int[primes.length];Arrays.fill(p, 1);long values[] = new long[primes.length];for (int i = 2; i <= n; i++) {long min = Long.MAX_VALUE;for (int j = 0; j < p.length; j++) {values[j] = primes[j] * dp[p[j]];min = Math.min(min, values[j]);}dp[i] = min;for (int j = 0; j < p.length; j++) {if (min == values[j]) {p[j]++;}}}return (int)dp[n];}public static void main(String[] args) {Solution solution = new Solution();System.out.println(solution.nthSuperUglyNumber(10, new int[]{2, 3, 5}));}
}

总结

从上面第一个问题的两个解决思路,可分为传统思路和动态规划,动态规划在耗时方面真实太优秀了;但是要想到却很费脑子,我主要还是看了别人的解题思路后,自己再实现了一遍,对动态规划这块一直做的不太好。如果有更高效、更简洁的解决方案欢迎回复。

【数学】丑数II 和 超级丑数相关推荐

  1. 找出落单的元素 java,lintcode练习 - 落单的数(落单的数 落单的数 II 落单的数 III)...

    落单的数 给出2*n + 1 个的数字,除其中一个数字之外其他每个数字均出现两次,找到这个数字. 样例 给出 [1,2,2,1,3,4,3],返回 4 挑战 一次遍历,常数级的额外空间复杂度 解题思路 ...

  2. 落单的数(落单的数 落单的数 II 落单的数 III)

    落单的数 给出2*n + 1 个的数字,除其中一个数字之外其他每个数字均出现两次,找到这个数字. 样例 给出 [1,2,2,1,3,4,3],返回 4 挑战  一次遍历,常数级的额外空间复杂度 思路1 ...

  3. 阿里云天池超级码力在线编程大赛初赛 第2场 ABCD(A.计算几何 判断点在三角形内 D.大施罗德数/超级卡特兰数)

    心得 打了一下被群友吐槽的比赛,阅读体验极差 阴间题面,读题1小时,AC5min,原题警告 思路来源 https://blog.csdn.net/PleasantlY1/article/details ...

  4. 【算法讲11:卡特兰数】默慈金数 | 那罗延数 | 施罗德数

    [算法讲11:卡特兰数]默慈金数 | 那罗延数 | 施罗德数 ⌈\lceil⌈卡特兰数⌋\rfloor⌋Catalan Number 引入 思考 ⌈\lceil⌈卡特兰数⌋\rfloor⌋的性质 ⌈\ ...

  5. LeetCode 313. 超级丑数(动态规划)

    1. 题目 编写一段程序来查找第 n 个超级丑数. 超级丑数是指其所有质因数都是长度为 k 的质数列表 primes 中的正整数. 示例: 输入: n = 12, primes = [2,7,13,1 ...

  6. LeetCode 263. 丑数 264. 丑数 II(DP)

    文章目录 1. LeetCode 263. 丑数 解题 2. LeetCode 264. 丑数 II DP解题 1. LeetCode 263. 丑数 编写一个程序判断给定的数是否为丑数. 丑数就是只 ...

  7. Leetcode 313. 超级丑数 解题思路及C++实现

    解题思路: 与Leetcode 264. 丑数 II的解题思路一样,均使用最小堆来存储丑数,第i次更新最小堆时,得到第i大的丑数. 可结合Leetcode 264. 丑数 II的解题思路理解:http ...

  8. 从C语言的角度重构数据结构系列(八)-数据结构堆知识超级丑数

    前言 在这里给自己打个广告,需要的小伙伴请自行订阅. python快速学习实战应用系列课程 https://blog.csdn.net/wenyusuran/category_2239261.html ...

  9. LeetCode 264. 丑数 II

    264. 丑数 II Ideas 竟然没想到用小根堆,白学了,再把小根堆抄一遍. Code Python class Solution:def nthUglyNumber(self, n: int) ...

最新文章

  1. 基本拖拽效果,使用 mousedown , mousemove , mouseup实现
  2. ASP.NET Identity “角色-权限”管理 2
  3. Django去操作已经存在的数据库
  4. 如何在windows中使用cmd命令去编译,运行C++程序
  5. Android入门笔记05
  6. GOOGLE本地搜索
  7. 软考 - 网络工程师考试大纲
  8. ElementUI表单构建
  9. mysql数据库行列矩阵调换位置(行与列调换)
  10. java短信接口_短信接口-- java
  11. 电脑键盘出现计算机,技巧:如何恢复计算机键盘上的乱码[设置方法]
  12. 怎样给CSDN博客添加微信二维码
  13. 今日头条推荐算法详解(PDF下载)
  14. 谷歌又闹大乌龙!Jeff Dean参与的新模型竟搞错Hinton生日
  15. 网址URL转义,特殊字符编码HTTP URL编码表
  16. 北航计算机是啥水平,清华眼中的北航是什么水平?附北航2020年全国录取分数线...
  17. 西门子HMI下载时提示缺少面板映像解决方案
  18. Lazada打造爆款秘籍
  19. 你不知道Python多能干,小伙子骨骼惊奇不来学习吗?
  20. WINSYS DAY02: 用户和组账号 、 NTFS权限控制

热门文章

  1. 优秀生被批评后“自闭”一个月 挫折教育不可少
  2. PyTorch学习笔记:nn.Sigmoid——Sigmoid激活函数
  3. 2年5个月13天,从外包到拿下阿里offer,没想到屌丝也能有今天
  4. 计算机科学与技术大学博士后,清华大学博士后
  5. 密钥ssh 配置操作
  6. 易启秀20150629完整包微场景制作源码,新增1.4G素材包,全新后台UI设计+采集
  7. python输出假分数_解析ArcGis的标注(一)——先看看分数式、假分数式标注是怎样实现的...
  8. 支付宝支付接口开发———在线扫码支付
  9. 关于订单扫码在线支付
  10. 164. Maximum Gap 1