剑指offer面试题——丑数

题目链接见：丑数

题目描述

把只包含因子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];
    }
};