33.N-th_Ugly_Number
2024年2月2日小于 1 分钟约 221 字
// 把只包含质因子2、3和5的数称作丑数(UglyNumber)
// 例如6、8都是丑数,但14不是,因为它包含质因子7。
// 习惯上我们把1当做是第一个丑数。求按从小到大的顺序的第N个丑数。
#include <algorithm>
#include <vector>
using namespace std;
class Solution {
public:
int GetUglyNumber_Solution(int index) {
if (index < 7) return index;
// p2,p3,p5分别为三个队列的指针,newNum为从队列头选出来的最小数
int p2 = 0, p3 = 0, p5 = 0, newNum = 1;
vector<int> arr;
arr.push_back(newNum);
while (arr.size() < index) {
//选出三个队列头最小的数
newNum = min(arr[p2] * 2, min(arr[p3] * 3, arr[p5] * 5));
//这三个if有可能进入一个或者多个,进入多个是三个队列头最小的数有多个的情况
if (arr[p2] * 2 == newNum) p2++;
if (arr[p3] * 3 == newNum) p3++;
if (arr[p5] * 5 == newNum) p5++;
arr.push_back(newNum);
}
return newNum;
}
};