开启掘金成长之旅!这是我参与「掘金日新计划 · 2 月更文挑战」的第 9 天,点击查看活动详情
一、题目描述
我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:
1 是丑数。
n 不超过1690。
二、思路讲解
大家对此有没有思路呢?我作为一个算法小白,一开始也是没有头绪的,多亏了大佬的提点和知道,后续才慢慢有了思路:
首先可以想到,所有的丑数都是由前面的丑数乘上2、3或5得来的。而我们需要有顺序的丑数序列,就有必要知道,前面的数乘上2、3或5之后的数谁大谁小。因此,我们可以设置三个指针a、b、c;指针a表示所指的数还没有乘2,指针b表示所指的数还没有乘3,指针c表示所指的数还没有乘5。那么下一个丑数就应该是min{dp[a]*2, dp[b]*3, dp[c]*5},然后对应的指针后移。比如说,如果最小的值为dp[a]*2,那么下一个丑数就是dp[a]*2,那么a指针就后移(移到没有乘过2的数上面,也就是下一个数),然后dp[i-1]即为第i个丑数。
三、Java代码实现
class Solution {
public int nthUglyNumber(int n) {
int []dp = new int[2000];
dp[0] = 1;
int a=0, b=0, c=0;
int count = 1;
while(count <= n){
int temp = Math.min(Math.min(dp[a]*2, dp[b]*3), dp[c]*5);
dp[count++] = temp;
//这里需要特别注意,有可能出现该丑数既等于dp[a]*2又等于dp[b]*3的情况
//此时a指针和b指针都应该后移,所以不能写成else if,必须是三个if
if(temp == dp[a]*2){
a++;
}
if(temp == dp[b]*3){
b++;
}
if(temp == dp[c]*5){
c++;
}
}
return dp[n-1];
}
}
四、时空复杂度分析
时间复杂度: O(N) 需要遍历dp数组
空间复杂度: O(N) 需要一个dp数组