题目:
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
思路:
我们可以依次写出丑数1,2,3,4,5,6,8,9,10,12,15,16,18,20......
仔细观察我们发现因为丑数只包含2,3,5三个因子,因此“下一个丑数”可以由前面的某一个丑数乘上这三个因子的某一个得到,前面丑数和因子的选择要保证两个要求:
- 得到的这个丑数比当前的所有丑数大,
- 比当前所有丑数大的丑数中最小的一个
比如当前的丑数系列到了20,那么下一个丑数由之前的某一个丑数加上一个因子2,3或者5得到;
- 如果是加上2这个因子,那么要保证上面两个要求,那么应该选择丑数12与2相乘,为24;
- 如果是加上3这个因子,那么要保证上面两个要求,那么应该选择丑数8与3相乘,为24;
- 如果是加上5这个因子,那么要保证上面两个要求,那么应该选择丑数5与5相乘,为24;
我们还要从这三个数中选最小的一个为24;24下一个丑数呢,依照上面的规则;
- 对于因子2,我们只需要从12之后的丑数开始看即可,12*2 = 24 = 24 ,15*2 = 30>24;
- 对于因子3:8*3 = 24 = 24,9*3 = 27 > 24;
- 对于因子5;5*5 = 25 > 24;
故最小的25为下一个丑数
具体操作
我们可以使用动态规划的方法,知道第一个丑数到第index-1个丑数,存储在数组res中,求第index个丑数;
定义三个整型数n1,n2,n3,分别记录满足res[n1]*2(res[n2]*3,res[n3]*5) > 第index - 2个丑数的res的最小索引;
从n1,n2,n3当前值出发向后搜索,直到找到能够使相应运算(res[n1]*2,res[n2]*3,res[n3]*5)大于第index - 1个丑数的新的索引n1,n2,n3;
上面实际上找到了3个丑数,其中最小的即为第index个丑数
代码
class Solution {
public:
int GetUglyNumber_Solution(int index) {
if(index <= 0)return 0;
if(index == 1)return 1;
vector<int> res(index);
res[0] = 1;
int n1 = 0,n2 = 0,n3 = 0;
for(int i = 1; i < index; i ++){
while(2*res[n1] <= res[i - 1])n1 ++;
while(3*res[n2] <= res[i - 1])n2 ++;
while(5*res[n3] <= res[i - 1])n3 ++;
res[i] = min(2*res[n1],min(3*res[n2],5*res[n3]));
}
return res[index - 1];
}
};