lintcode4-4. 丑数 II

设计一个算法，找出只含素因子2，3，5 的第 n 小的数。

符合条件的数如：1, 2, 3, 4, 5, 6, 8, 9, 10, 12...

 注意事项

我们可以认为1也是一个丑数

您在真实的面试中是否遇到过这个题？ 

Yes

样例

如果n = 9， 返回 10

挑战 

要求时间复杂度为O(nlogn)或者O(n)

标签 

LintCode 版权所有  优先队列

解析：暴力解法不做介绍大家都可以想到，在该题会time limited

换一种思路是：所有的肯定都是由2，3，5这三个数组成（1不算），则可以理解为后面的数由前面的数与2，3，5组合而成的，下面的代码重点是如何找到下一个最小的数，具体的做法可以在代码中看到就不多做解释。

class Solution {
public:
    /*
     * @param n: An integer
     * @return: the nth prime number as description.
     */
    int nthUglyNumber(int n) {
        // write your code here
        vector<int> uglyNum;
        int num2=0,num3=0,num5=0;
        uglyNum.push_back(1);
        for(int i=1;i<n;i++){
            int min1=uglyNum[num2]*2>uglyNum[num3]*3?uglyNum[num3]*3:uglyNum[num2]*2;
            int min2=min1>uglyNum[num5]*5?uglyNum[num5]*5:min1;
            uglyNum.push_back(min2);
            cout<<min2<<endl;
            if(uglyNum[i]/2==uglyNum[num2])
                num2++;
            if(uglyNum[i]/3==uglyNum[num3])
                num3++;
            if(uglyNum[i]/5==uglyNum[num5])
                num5++;
        }
        return uglyNum.back();
    }
};