题目描述
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
思路:动态规划
维护一个存储有前n个丑数的队列
首先初始队列中只有第一个丑数1
1的下一个的丑数 就是 1*2 1*3 1*5 中最小的那个
2最小,那么把2加进丑数队列之后
队列变成{1, 2}
队列的下一个丑数,就是1和2分别乘以2 、3 、5中的最小值
为了方便计算,在乘以2、3、5时,我们得知道是1还是2来乘以这三个数时有最小值
那么可以设置3个数,保存乘以2、3、5有最小值的队列元素的下标
每需要加进一个新的丑数时,就动态的调整这些下标值,直到第n个丑数出现
class Solution {
public:
int GetUglyNumber_Solution(int index) {
if(index <= 0) return 0;
int p2 = 0,p3 = 0,p5 = 0;
vector<int> ugly;
int m = 1;
ugly.push_back(m);
while(ugly.size() != index){
//已有丑数队列中元素 乘以 2 3 5中最小的数
m = min(ugly[p2] * 2,min(ugly[p3] * 3,ugly[p5] * 5));
if(m == ugly[p2] * 2) p2++;
if(m == ugly[p3] * 3) p3++;
if(m == ugly[p5] * 5) p5++;
ugly.push_back(m);
}
return m;
}
};