题目描述
把只包含质因子2、3和5的数称作丑数(Ugly Number)。例如6、8都是丑数,但14不是,因为它包含质因子7。 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
思路
第一个丑数是1,其实之后的丑数是之前的一个丑数的2、3、5倍,按顺序求得话,就得把他们的2倍、3倍、5倍的最小值来加入到存储丑数的数组中,直至求得的丑数的数量足够。
C++
class Solution {
public:
int GetUglyNumber_Solution(int index) {
if (index <= 0) return 0;
vector<int>res; //存储丑数的数组
int mul2 = 0; //数组最后一个2的倍数的所在位置
int mul3 = 0; //数组最后一个3的倍数的所在位置
int mul5 = 0; //数组最后一个5的倍数的所在位置
res.push_back(1);
int count = 1;
int min;
while (count < index)
{
//每一个丑数都是前面丑数的2倍、3倍、或5倍
min = minchou(res[mul2] * 2, res[mul3] * 3, res[mul5] * 5);
res.push_back(min);
if (res[mul2] * 2 <= min) mul2++;
if (res[mul3] * 3 <= min) mul3++;
if (res[mul5] * 5 <= min) mul5++;
count++;
}
return res[index - 1];
}
int minchou(int x, int y, int z)
{
int min = x < y? x:y;
min = min < z? min:z;
return min;
}
};
Python
# -*- coding:utf-8 -*-
class Solution:
def GetUglyNumber_Solution(self, index):
# write code here
if index <= 0:
return 0
res = []
res.append(1)
mul2 = mul3 = mul5 = 0
count = 1
while count < index:
minchou = min(res[mul2] * 2, res[mul3] * 3, res[mul5] * 5)
res.append(minchou)
if res[mul2] * 2 <= minchou:
mul2 += 1
if res[mul3] * 3 <= minchou:
mul3 += 1
if res[mul5] * 5 <= minchou:
mul5 += 1
count += 1
return res[index - 1]
参考
https://blog.csdn.net/qq_42236766/article/details/98620238