题目:我们把只包含因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含因子7。习惯上我们把1当做是第一个丑数。求出第n个丑数
最容易想到的是从1开始依次判断是否为丑数并计数,但基于丑数的增长速度快当n较大时程序的运行效率会非常缓慢。
我利用丑数必定是若干个2,3,5的乘积,维护一个顺序的丑数数组,而这个数组的第n个元素即是第n个丑数。
假设数组a(大小为n)包含从第一个到第n个的所有丑数,则按照丑数的定义,第n+1个丑数必定是a中的某个丑数乘以2或3或5。
最容易想到的是从1开始依次判断是否为丑数并计数,但基于丑数的增长速度快当n较大时程序的运行效率会非常缓慢。
我利用丑数必定是若干个2,3,5的乘积,维护一个顺序的丑数数组,而这个数组的第n个元素即是第n个丑数。
假设数组a(大小为n)包含从第一个到第n个的所有丑数,则按照丑数的定义,第n+1个丑数必定是a中的某个丑数乘以2或3或5。
那么就可以由1~n的丑数求出第n+1个丑数了,依次类推可以从第一个丑数一直求到第n个,解法如下:
#include <vector>
#include <iostream>
#include <time.h>
using namespace std;
int main()
{
unsigned int n;
cin>>n;
time_t timebegin = time(NULL);
vector<double> vec;
vec.reserve(n);
vec.push_back(1);
vec.push_back(2);
vec.push_back(3);
vec.push_back(4);
vec.push_back(5);
unsigned int t2=0;
unsigned int t3=0;
unsigned int t5=0;
while(vec.size()<n)
{
int tempsize=vec.size();
int pos=vec.size()-1;
double min=vec[pos]*2;
for(int i=t2;i<tempsize;i++)
{
if(vec[i]*2>vec[vec.size()-1])
{
if(vec[i]*2<min)
{
min=vec[i]*2;
t2=i;
}
break;
}
}
for(int i=t3;i<tempsize;i++)
{
if(vec[i]*3>vec[vec.size()-1])
{
if(vec[i]*3<min)
{
min=vec[i]*3;
t3=i;
}
break;
}
}
for(int i=t5;i<tempsize;i++)
{
if(vec[i]*5>vec[vec.size()-1])
{
if(vec[i]*5<min)
{
min=vec[i]*5;
t5=i;
}
break;
}
}
vec.push_back(min);
}
cout<<vec[n-1]<<endl;
cout<<"time spent: "<<time(NULL)-timebegin<<endl;
return 0;
}