LeetCode-剑指offer2-49-丑数
一、题目描述
我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数
根据题意, 一个丑数乘以 2, 3, 5 之后, 一定还是一个丑数。但题目关键在于如何将丑数有序输出。
并且如果从丑数序列首个元素开始 *2 *3 *5 来计算, 丑数序列也是不会产生漏解的。
二、解题思路
1.暴力法
我们只需要从1开始不断的乘2,乘3,乘5进行穷举到极限范围,然后对数组进行排序,返回对于的值就好。
class Solution {
public:
int nthUglyNumber(int n) {
List<Integer> ugly = new ArrayList<>();
for(long i = 1; i < INT_MAX; i *= 2){
for(long j = i; j < INT_MAX; j *= 3){
for(long k = j; k < INT_MAX; k *= 5){
ugly.push_back(k);
}
}
}
ugly.sort();
return ugly[n - 1];
}
};
2.动态规划
2.1丑数的排列
假设当前存在 3 个数组 nums2, nums3, nums5 分别代表扩展丑数序列num从 1 开始分别乘以 2, 3, 5 的序列, 即
丑数序列 num = {1, 2, 3, 4, 5, 6, 8, 9, 10, 12 }
nums2 = {1*2, 2*2, 3*2, 4*2, 5*2, 6*2, 8*2...}
nums3 = {1*3, 2*3, 3*3, 4*3, 5*3, 6*3, 8*3...}
nums5 = {1*5, 2*5, 3*5, 4*5, 5*5, 6*5, 8*5...}
2.2.有序化扩展丑数数组
从丑数的排列,我们可以发现,最终的丑数序列实际上就是这 3 个有序序列对的合并结果, 计算丑数序列也就是相当于 合并 3 个有序序列。同时对相同的值进行去重。
最简单的方法就是每一个序列都各自维护一个指针, 然后比较指针指向的元素的值, 将最小的放入最终的合并数组中, 并将相应都相等的指针向后移动一个元素。
// 1, 2, 3, 4, 5, 6, 8, 9, 10, 12
class Solution {
public:
int nthUglyNumber(int n) {
vector<int> dp(n, 0);
dp[0] = 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];
}
};