我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:
- 1 是丑数。
- n 不超过1690。
分析:
方法:动态规划
令第 n 个丑数为 Fn,那么 Fn 一定是 2、3、5 中一个或多个数的倍数,令这个倍数为 a、b、c,那么 Fn 就是 2a、3b、5c 中的一个数,因为 Fn 只包含 2、3、5 的质因子,所以 a、b、c 也是丑数,那么 a、b、c 就可以看做 Fi,Fj,Fk,因为丑数是升序排列的,所以 Fn 一定是 2Fi、3Fj、5Fk 中的最小值,找到 Fn 的值后,我们再将最小值的索引加1,这样就可以继续计算 Fn+1 的值,因此我们可以写出转移方程:
dp[ n ] = min{ 2*dp[ i ],3*dp[ j ],5*dp[k] } ( i, j, k >= 0 )
时间复杂度:O(n)
空间复杂度:O(n)
class Solution {
public int nthUglyNumber(int n) {
//初始化索引
int i = 0, j = 0, k = 0;
//初始化结果数组
int[] dp = new int[n];
//1是第一个丑数
dp[0] = 1;
//遍历,赋值
for(int m = 1; m < n; ++m){
//当前可能的值
int a = dp[i] * 2, b = dp[j] * 3, c = dp[k] * 5;
//取最小值
dp[m] = Math.min(a, Math.min(b, c));
//最小值索引加1,可能不止一个最小值,所以不用if-else
//a
if(dp[m] == a){
i++;
}
//b
if(dp[m] == b){
j++;
}
//c
if(dp[m] == c){
k++;
}
}
//返回对应位置丑数
return dp[n-1];
}
}
题目来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/chou-shu-lcof