问题描述:
设计一个算法,找出只含素因子2,3,5 的第 n 小的数。
符合条件的数如:1, 2, 3, 4, 5, 6, 8, 9, 10, 12…
样例
如果n = 9, 返回 10
挑战
要求时间复杂度为O(nlogn)或者O(n)
注意事项
我们可以认为1也是一个丑数
方法一:可以遍历每一个数,当它是丑数时,计数加1,直到它是第n个丑数。
而丑数的判断是:2^i *3^j *5*k。因此。首先整除2,直到不能整除为止,然后整除3,直到不能整除为止,然后整除5,直到不能整除为止。若是丑数,则能依次整除,那么最后得到的结果是1;如果不是丑数,则不能整除完,得到的结果不是1。
该方法所需时间较长,因为是遍历每个整数,不是丑数的也进行了整除判断。代码如下:
class Solution{
public:
int nthuglynumber(int n){
int count=0;
int m=0;
int number;
while(count<n){
m++;
number=m;
while(number%2==0)
number/=2;
while(number%3==0)
number/=3;
while(number%5==0)
number/=5;
if(number==1)
count++;
}
return m;
}
方法二:该方法是想由较小的丑数得到后面的丑数,这样得到一个丑数数组,其中的丑数是按从小到大的顺序排列的。从而跳过非丑数的判断,节省了时间,但由于要创建一个n个元素的数组,浪费了空间。具体思路为:
首先定义一个数组存放丑数,认为1是丑数,则初始化数组num[0] = 1,然后从2,3,5这三个种子中挑选,选择num[0]*2,num[0]*3,num[0]*5中最小的数为新的丑数,显然应该选择2,即num[1] = 2,然后从num[1]*2,num[1]*3,num[1]*5,num[0]*3,num[0]*5中进行选择,由于数组时按从小到大排列的,显然只用考虑num[1]*2,num[0]*3,num[0]*5,而不用考虑num[1]*3,num[1]*5,选择3,即num[2] = 3,依次进行操作。我们可以看到,当*2被选出了一次,则下次进行比较的时候,它的idx_2就加一,对*3和*5一样,这样相当于将所有已经得到的丑数都乘以了2 3 5,然后比较最小值,只是每次只需要三个数进行比较,因为同一个*2的丑数,根据idx_2就知道它们的大小。 代码如下:
class Solution {
public:
/**
* @param n: An integer
* @return: the nth prime number as description.
*/
int nthUglyNumber(int n) {
int *ugly=new int[n];
ugly[0]=1;
int idx_2=0;
int idx_3=0;
int idx_5=0;
for(int i=1;i<n;i++){
ugly[i]=min(min(ugly[idx_2]*2,ugly[idx_3]*3),ugly[idx_5]*5);
if(ugly[i]==(ugly[idx_2]*2))
idx_2++;
if(ugly[i]==(ugly[idx_3]*3))
idx_3++;
if(ugly[i]==(ugly[idx_5]*5))
idx_5++;
}
int result=ugly[n-1];
delete ugly;
return result;
}
};