丑数II - LintCode

描述
写一个程序来检测一个整数是不是丑数。

丑数的定义是，只包含质因子 2, 3, 5 的正整数。比如 6, 8 就是丑数，但是 14 不是丑数以为他包含了质因子 7。

可以认为 1 是一个特殊的丑数。

样例
给出 num = 8，返回 true。
给出 num = 14，返回 false。

思路
构建数组nums存放丑数，nums[0]=1。将当前的丑数当做之前求得的丑数乘以因子后得到的最小值。num_2，num_3，num_5分别表示乘以2，3，5的丑数的位置。最开始位置都是0，选取nums[num_2]*2，nums[num_3]*3，nums[num_5]*5的最小值作为当前丑数的值。并更新num_2，num_3，num_5的值。

#ifndef C004_H
#define C004_H
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
class Solution {
public:
    /**
    * @param n: An integer
    * @return: the nth prime number as description.
    */
    int nthUglyNumber(int n) {
        // write your code here
        if (n <= 0)
            return 0;
        //每个丑数都可以用2^i*3^j*5^k的形式来表示,将其排序，得到的第n个即为结果
        vector<int> nums(n, 0);
        nums[0] = 1;
        int num_2 = 0, num_3 = 0, num_5 = 0;
        for (int i = 1; i < n; ++i)
        {
            nums[i] = min(min(nums[num_2] * 2, nums[num_3] * 3), nums[num_5] * 5);
            if (nums[i] % 2 == 0)
                num_2++;
            if (nums[i] % 3 == 0)
                num_3++;
            if (nums[i] % 5 == 0)
                num_5++;
        }
        return nums[n - 1];
    }
};
#endif