题目描述:我们把只包含因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:
1 是丑数。
n 不超过1690。
解题思路:前20个丑数为:1, 2, 3, 4, 5, 6, 8, 9, 10, 12, 15, 16, 18, 20, 24, 25, 27, 30, 32, 36。
通过规律,可以得出后一个数字为前某个数字 乘以2 或者 3 或者 5,取其中最小值。
- 定义3个指针 p2、 p3、 p5 分别指向要乘以 2 、3、5的数字,dp[ i ]表示第 i 个丑数。
- dp[p2] * 2、dp[p3] * 3、dp[p5] * 5的最小值就是下一个丑数。
- 如果dp[ i ] 与dp[p2] * 2、dp[p3] * 3、dp[p5] * 5中任意一个相等,那么相等的index++
class Solution {
public:
int nthUglyNumber(int n) {
vector<int> dp(n,0); //创建n个元素,初始化为0
dp[0] = 1; // 第一个丑数为 1
int p2 = 0, p3 = 0, p5 = 0;
for(int i = 1; i < n; ++i)
{
dp[i] = min(min(dp[p2] * 2 , dp[p3] * 3), dp[p5] * 5);
if(dp[i] == dp[p2] * 2) p2++;
if(dp[i] == dp[p3] * 3) p3++;
if(dp[i] == dp[p5] * 5) p5++;
}
return dp[n - 1];
}
};