Question
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
关键词:丑数 最小指针 排序
Solution
递归
丑数的本质是a个2b个3c个5,也可以看成是已有的某个丑数2、3、5得到新的丑数,本质是无重复的排序问题。如果是每个丑数都乘一边,得到:1,2,3,5,22=4,23=6, 25=10,32=6,33=9…可见不仅会出现重复,而且得到的序列不是有序的。利用三个指针分别指向三个丑数a、b、c,下一步只算三个新丑数a2、b3、c5,取最小值为下一个丑数(保证序列有序),其次是只要新丑数被取到,对应指针+1,比如在a=3,b=2,c=2的情况下,最小新丑数为6,既可能是a2取到的也可能是b*3取到,这时a、b指针都要+1(保证无重复,丑数6不会再出现)。
时间复杂度:O(N)
空间复杂度:O(N)
- Python
class Solution:
def GetUglyNumber_Solution(self, index):
if index<2:
return index
save = [1]
a, b, c = 0, 0, 0
for i in range(1, index):
cur_min = min(save[a]*2, save[b]*3, save[c]*5)
if cur_min==save[a]*2:
a += 1
if cur_min==save[b]*3:
b += 1
if cur_min==save[c]*5:
c += 1
save.append(cur_min)
return cur_min
- C++
class Solution {
public:
int GetUglyNumber_Solution(int index) {
if(index<2)
return index;
vector<int> save = {1};
int a = 0, b= 0, c = 0;
int cur_min = 1;
for (int i=1; i<index; i++){
cur_min = min(save[a]*2, min(save[b]*3, save[c]*5));
save.push_back(cur_min);
// 都是if保证所有重复丑数对应的指针全部更新
if(cur_min == save[a]*2)
a++;
if(cur_min == save[b]*3)
b++;
if(cur_min == save[c]*5)
c++;
}
return cur_min;
}
};