描述
写一个程序来检测一个整数是不是丑数。
丑数的定义是,只包含质因子 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