题目链接见:丑数
题目描述
把只包含因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
分析:首先的思路是暴力搜索方法,但是这样存在的问题就是效率太低,很容易超时。具体有两种实现方式:(1)先将一些丑数存储在一个容器中,再对容器进行排序,最后取出第N个丑数即可;(2)从开始的数一个个进行判断,到第N个停止即可。代码如下:
(1)
class Solution {
public:
int GetUglyNumber_Solution(int index) {
vector<long long> v;
for(int i = 0;i<11;++i)
for(int j = 0;j<11-i;++j)
for(int k = 0;k<11-i-j;++k)
v.push_back(pow(2,i)*pow(3,j)*pow(5,k));
sort(v.begin(),v.end());
return v[index-1];
}
};
(2)
class Solution {
public:
bool isugly(int n){
if(n==1)
return true;
while(n%2==0||n%3==0||n%5==0){
if(n%2==0)
n/=2;
if(n%3==0)
n/=3;
if(n%5==0)
n/=5;
}
return (n==1)?true:false;
}
int GetUglyNumber_Solution(int index) {
int st = 1,cnt = 0;
while(cnt<index){
if(isugly(st))
++cnt;
++st;
}
return st-1;
}
};
换一种思路,我们可以建立一个数组,将丑数全部存储在数组里。那么数组中靠后的丑数可以由前面的丑数通过乘以2,3,5比较得到。这是动态规划的思想。具体代码如下:
class Solution {
public:
int findmin(int a,int b,int c){
int tmp = (a<b)?a:b;
return (tmp<c)?tmp:c;
}
int GetUglyNumber_Solution(int index) {
int *arr = new int[index+1];
arr[0] = 1;
arr[1] = 2;
arr[2] = 3;
arr[3] = 4;
arr[4] = 5;
for(int i = 5;i<index;++i){
int min1 = 0,min2 = 0,min3 = 0;
for(int j = 0;j<i;++j){
int tmp = arr[j]*2;
if(tmp>arr[i-1]){
min1 = tmp;
break;
}
}
for(int j = 0;j<i;++j){
int tmp = arr[j]*3;
if(tmp>arr[i-1]){
min2 = tmp;
break;
}
}
for(int j = 0;j<i;++j){
int tmp = arr[j]*5;
if(tmp>arr[i-1]){
min3 = tmp;
break;
}
}
arr[i] = findmin(min1,min2,min3);
}
return arr[index-1];
}
};