leetcode264.丑数2

题目大意

编写一个程序,找出第 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();
	}
};

猜你喜欢

转载自blog.csdn.net/qq_41092190/article/details/105943392