题目链接:戳这里~
题目描述:
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
思路:
由于丑数的特殊性质,所有的丑数都是前面某一个丑数的2或3或5倍,即如果要求第 n 个丑数,只需求前 n - 1 个丑数的2、3、5倍,找到前 n - 1 个丑数中乘以2、3、5后刚刚好大于第 n - 1 个丑数的那个数,这个数就是我们要求的第 n 个丑数。
代码实现:
1. Java
class Solution {
public int GetUglyNumber_Solution(int index) {
if(index == 0) return 0;// 第 0 个丑数为 0
int[] a = new int[index + 1]; // 用于存放已找到的丑数
a[0] = 1;
for(int i = 0 ; i < index ; i++){
int tempUgly2 = 0, tempUgly3 = 0, tempUgly5 = 0; //这三个变量依次存放 丑数数组前 i 个丑数分别乘以2/3/5后的值
for(int j = 0 ; j <= i ; j++){
if(a[j] * 2 > a[i]){
tempUgly2 = 2 * a[j];
break;
}
}
for(int j = 0 ; j <= i ; j++){
if(a[j] * 3 > a[i]){
tempUgly3 = 3 * a[j];
break;
}
}
for(int j = 0 ; j <= i ; j++){
if(a[j] * 5 > a[i]){
tempUgly5 = 5 * a[j];
break;
}
}
int temp = tempUgly2 < tempUgly3 ? tempUgly2 : tempUgly3;
a[i+1] = temp < tempUgly5 ? temp : tempUgly5;
}
return a[index - 1]; // 由于数组下标从 0 开始,所以返回的是 index - 1
}
}
2. C++ (某大佬的代码),很简洁
链接:https://www.nowcoder.com/questionTerminal/6aa9e04fc3794f68acf8778237ba065b
来源:牛客网
class Solution {
public://别人的代码就是精简,惭愧啊,继续学习。
int GetUglyNumber_Solution(int index) {
if (index < 7)return index;
vector<int> res(index);
res[0] = 1;
int t2 = 0, t3 = 0, t5 = 0, i;
for (i = 1; i < index; ++i)
{
res[i] = min(res[t2] * 2, min(res[t3] * 3, res[t5] * 5));
if (res[i] == res[t2] * 2)t2++;
if (res[i] == res[t3] * 3)t3++;
if (res[i] == res[t5] * 5)t5++;
}
return res[index - 1];
}
};