1、题目描述
我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
2、VS2019上运行
使用方法二:动态规划
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
class Solution {
public:
int nthUglyNumber(int n) {
vector<int> dp(n + 1); // 动态数组,存储丑数
dp[1] = 1; // 第一个丑数为1
int p2 = 1, p3 = 1, p5 = 1; // 三个指针,指向下一个乘以2、3和5的位置
for (int i = 2; i <= n; i++) {
int num2 = dp[p2] * 2, num3 = dp[p3] * 3, num5 = dp[p5] * 5; // 计算三个候选丑数
dp[i] = min(min(num2, num3), num5); // 选择最小的丑数作为dp[i]
if (dp[i] == num2) {
p2++; // 移动p2指针到下一个位置
}
if (dp[i] == num3) {
p3++; // 移动p3指针到下一个位置
}
if (dp[i] == num5) {
p5++; // 移动p5指针到下一个位置
}
}
return dp[n]; // 返回第n个丑数
}
};
int main() {
Solution s;
int n = 10;
int ans = s.nthUglyNumber(n);
cout << "The " << n << "th ugly number is: " << ans << endl;
return 0;
}
The 10th ugly number is: 12
3、解题思路
- 1.创建一个动态数组 dp,用于存储生成的丑数,其中 dp[1] 初始化为 1,表示第一个丑数为 1。
- 2.创建三个指针 p2、p3 和 p5,分别表示下一个将乘以 2、3 和 5 的丑数在 dp 数组中的位置。
- 3.从 i = 2 开始迭代到 n:
- 4.计算三个候选丑数:num2 = dp[p2] * 2,num3 = dp[p3] * 3,num5 = dp[p5] * 5。
- 5.将最小的候选丑数赋值给 dp[i],即 dp[i] = min(min(num2, num3), num5)。
- 6.如果 dp[i] 等于 num2,则将 p2 指针向后移动一个位置,即 p2++。
- 7.如果 dp[i] 等于 num3,则将 p3 指针向后移动一个位置,即 p3++。
- 8.如果 dp[i] 等于 num5,则将 p5 指针向后移动一个位置,即 p5++。
- 9.循环结束后,dp[n] 即为第 n 个丑数,将其作为结果返回。