剑指Offer(第2版)——面试题49：丑数

题目

把只包含质因子2、3和5的数称作丑数（Ugly Number）。例如6、8都是丑数，但14不是，因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。

质数是指在大于1的自然数中，除了1和它本身以外不再有其他因数的自然数。

解题思路

思路1：逐个判断

• 最简单的思路就是逐个判断每个整数是不是丑数。因为丑数只能被2，3，5整除。也就是说，如果一个数能被2整除，就继续除以2；能被3整除，就继续除以3；能被5整除，就继续除以5。如果最后得到的结果是1，那么这个数就是一个丑数。
• 最大的问题是每个整数都需要计算，即使一个数字不是丑数，我们也要对它执行求余数和除法操作，因此算法时间效率不高。

思路2：用空间换时间

• 根据丑数的定义，一个丑数应该是另一个丑数乘上2，3，5的结果（1除外）。因此我们可以创建一个数组，里面的数字是排好序的丑数，每一个丑数都是前面某个丑数乘以2、3或者5得到的。
• 这个思路的关键问题在于怎样保证数组里面的丑数是排好序的。我们把数组中现有的最大丑数记做M。现在我们来生成下一个丑数，该丑数肯定是前面某一个丑数乘以2、3或者5的结果。
• 对于乘以2而言，数组中一定存在某个丑数M2，排在它前面的每一个丑数乘以2得到的结果都会小于等于已知最大丑数M；M2是第一个乘2大于数组中最大丑数的那个数。我们只需要记下这个丑数的位置，同时每次生成新的丑数的时候，去更新这个M2。对乘以3和5而言，也存在着同样的M3和M5。
• 那么要添加到数组中的下一个丑数就是M2×2，M3×3，M5×5中最小的那一个。 按照这个思路向数组中添加丑数，直到找到需要的那个丑数为止。

代码

class Solution{
public:
    int GetUglyNumber_Solution(int index){
        // 1-6都是丑数 直接返回index
        if(index < 7) return index;
        vector<int> res(index);
        // 存放前6个丑数
        for(int i = 0; i < 6; i++){
            res[i] = i+1;
        }
        //当前最大丑数为6，在下标为5的位置上
        int m2 = 3, m3 = 2, m5 = 1;
        for(int i = 6; i < index; i++){
            res[i] = min(res[m2] * 2, min(res[m3] * 3, res[m5] * 5));
            while(res[i] >= res[m2] * 2) m2++;
            while(res[i] >= res[m3] * 3) m3++;
            while(res[i] >= res[m5] * 5) m5++;
        }
        return res[index - 1];
    }  
};

通用版：

class Solution{
public:
    int GetUglyNumber_Solution(int index){
        //错误输入
        if(index <= 0) return 0;
        //特殊输入
        if(index == 1) return 1;
        vector<int> res(index);
        res[0] = 1;
        //M2 = res[m2]
        int m2 = 0, m3 = 0, m5 = 0, i;
        for(i = 1; i < index; i++){
            res[i] = min(res[m2] * 2, min(res[m3] * 3, res[m5] * 5));
            if (res[i] == res[m2] * 2) m2++;
            if (res[i] == res[m3] * 3) m3++;
            if (res[i] == res[m5] * 5) m5++;
        }
        return res[index - 1];
    }  
};