题目描述
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
关键在于,这个题不能使用暴力法解,因为涉及到排除数的算法太复杂,如果使用暴力法,每一个数去取最小公因数,判断公约数是否只为2,3,5,这样每个数计算的时间为 ,判断N个数需要的时间为>N,所以会超时
改进算法,一个数只由2,3,5乘积组成,那么我们可以采取过滤法
使用 从 2 , 3 ,5 这个队列入手,
初步算法如下:
一、将队列中的每个数乘以 2, 3, 5 得到
2,6, 10
3 ,6, 15
5, 10,25
二、去掉重复数字,得到新队列
2,3,5,6,10,15,25
三、重复一、二操作
这三个队列,我们发现有重复的数字,这时设计算法,把重复的去掉,同时从小到大排序,到第N个即为所需丑数
代码优化版:
维护一个递增队列,和三个下标值,从小到大推进该三个下标值
代码实现如下:
class Solution {
public:
int GetUglyNumber_Solution(int index)
{
if (index == 0)
return 0;
std::vector<int> que;
que.resize(index);
que[0] = 1;
int i = 0;
int j = 0;
int k = 0;
int count = 1;
while (count < index)
{
int f = GetMinAToB(GetMinAToB(que[i] * 2, que[j] * 3), que[k] * 5);
if (f == que[i] * 2) i++;
if (f == que[j] * 3) j++;
if (f == que[k] * 5) k++;
que[count] = f;
count++;
}
return que[count - 1];
}
inline int GetMinAToB(int A, int B)
{
return A < B ? A : B;
}
};