题目大意
编写一个程序,找出第 n 个丑数。
丑数就是只包含质因数 2, 3, 5 的正整数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
解题思路
使用三指针,分别表示2、3、5产生的上一个丑数的索引,创建一个数组用来收集所有得到的丑数。
当前丑数等于:2/3/5指针对应的上一个丑数分别乘上自己的最小值。同时,判断一下这个最小值是哪几个因子的倍数,让该因子的指针向后移动一位。
class Solution {
public:
int nthUglyNumber(int n) {
if (n < 7)
return n;
vector<int>ugly{ 1 };
int count = 1, two = 0, three = 0, five = 0;
while (count < n)
{
// 找出三者中最小的值
int min1 = min(ugly[two] * 2, ugly[three] * 3);
int min2 = min(ugly[three] * 3, ugly[five] * 5);
int m = min(min1, min2);
ugly.push_back(m);
// 如果丑数是2的倍数,则2的指针向后移动
if (m == ugly[two] * 2)
++two;
// 如果丑数是3的倍数,则2的指针向后移动
if (m == ugly[three] * 3)
++three;
// 如果丑数是5的倍数,则2的指针向后移动
if (m == ugly[five] * 5)
++five;
++count;
}
return ugly.back();
}
};