37.寻找丑数[Ugly numbers]

【题目】

我们把只包含质因子2、3和5的数称作丑数(Ugly Number),例如:2,3,4,5,6,8,9,10,12,15,等,习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第1500个丑数。

【分析】

这是一道在网络上广为流传的面试题,据说google曾经采用过这道题。

所谓一个数m是另一个数n的因子,是指n能被m整除,也就是n % m == 0。根据丑数的定义,丑数只能被2、3和5整除。也就是说如果一个数如果它能被2整除,我们把它连续除以2;如果能被3整除,就连续除以3;如果能被5整除,就除以连续5。如果最后我们得到的是1,那么这个数就是丑数,否则不是。

基于前面的分析,我们可以写出如下的函数来判断一个数是不是丑数:

最普通(也最耗时)的做法是从1开始遍历,然后判断这个数的因式分解中只包含2,3,5,满足则找到了一个,一直找下去,直到第n个被找出!测试了一下,找第1500个丑数耗时40秒!

【方案1】

 C++ Code 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
 

bool IsUglyNumber(int n)
{
    
while(n % 2 == 0)
        n /= 
2;
    
while(n % 3 == 0)
        n /= 
3;
    
while(n % 5 == 0)
        n /= 
5;
    
return n == 1;
}

int GetUglyNumber_Solution1(int index)
{
    
if (index <= 0)
        
return 0;

    
int number = 0;
    
int uglyFound = 0;
    
while(uglyFound < index)
    {
        number++;
        
if (IsUglyNumber(number))
        {
            uglyFound++;
        }
    }
    
return number;
}

我们只需要在函数GetUglyNumber_Solution1中传入参数1500,就能得到第1500个丑数。该算法非常直观,代码也非常简洁,但最大的问题我们每个整数都需要计算。即使一个数字不是丑数,我们还是需要对它做求余数和除法操作。因此该算法的时间效率不是很高。

接下来我们换一种思路来分析这个问题,试图只计算丑数,而不在非丑数的整数上花费时间。根据丑数的定义,丑数应该是另一个丑数乘以2、3或者5的结果(1除外)。因此我们可以创建一个数组,里面的数字是排好序的丑数。里面的每一个丑数是前面的丑数乘以2、3或者5得到的。

假设数组ugly[N]中存放不断产生的丑数,初始只有一个丑数ugly[0]=1,由此出发,下一个丑数由因子2,3,5竞争产生,得到ugly[0]*2, ugly[0]*3, ugly[0]*5, 显然最小的那个数是新的丑数,所以第2个丑数为ugly[1]=2,开始新一轮的竞争,由于上一轮竞争中,因子2获胜,这时因子2对应的索引应该加1,变为1,得到ugly[1]*2,ugly[0]*3,ugly[0]*5, 因子3获胜,ugly[2]=3,同理,下次竞争时因子3对应的索引应该加1,变为1,即:ugly[1]*2, ugly[1]*3, ugly[0]*5, 因子5获胜,得到ugly[3]=5,重复这个过程,直到第n个丑数产生。总之:每次竞争中有一个(也可能是两个)因子胜出,下一次竞争中胜出的因子对应的索引应该加1!

方案2】

 

 C++ Code 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
 

int mymin(int a, int b, int c)
{
    
int tmp = a < b ? a : b;
    
int res = tmp < c ? tmp : c;
    
return res;
}

int GetUglyNumber_Solution2(int index)
{
    
if (index <= 0)
        
return 0;
    
int *ugly = new int[index];
    ugly[
0] = 1;
    
int index2 = 0;
    
int index3 = 0;
    
int index5 = 0;

    
int uglyIndex = 1;
    
while(uglyIndex < index)
    {
        
int minValue = mymin(ugly[index2] * 2, ugly[index3] * 3, ugly[index5] * 5);
        
// store ugly number
        ugly[uglyIndex] = minValue;
        uglyIndex++;

        
// find index which gets minValue, and then move it next
        if (minValue == ugly[index2] * 2)
            index2++;
        
if (minValue == ugly[index3] * 3)
            index3++;
        
if (minValue == ugly[index5] * 5)
            index5++;
    }
    
int result = ugly[index - 1];
    
delete []ugly;
    
return result;
}

 【参考】

http://zhedahht.blog.163.com/blog/static/2541117420094245366965/

http://www.cppblog.com/zenliang/articles/131094.html

http://www.cnblogs.com/coser/archive/2011/03/07/1976525.html