[剑指offer]丑数
题目描述
我们把只包含质因子 2、3 和 5 的数称作丑数(Ugly Number)。求按从小到大的顺序的第 n 个丑数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
说明:
1 是丑数。
n 不超过1690。
解题思路
我们知道,丑数的序列为1,2,3,4,5,6,8,9,10……,任何一个丑数都是由小于它的某个丑数×2、×3或×5得来的,假设有3个数组:
- Two:1,1×2,2×2,3×2,4×2,5×2,6×2,8×2,9×2,10×2,……
- Three:1,1×3,2×3,3×3,4×3,5×3,6×3,8×3,9×3,10×3,……
- Five:1,1×5,2×5,3×5,4×5,5×5,6×5,8×5,9×5,10×5,……
那么所有丑数的排列,必定是上面3个数组去重排序的结果,所以,就转换成了3个有序数组无重复元素合并的问题,注意这仨个数组是{1,2,3,4,5,6,8,9,10……}×2、×3、×5得到的。
想一下归并排序中的合并有序数组,每个数组对应一个指针,比较这3个指针对应的值,取最小值放入结果数组,然后该指针后移一位。
对应到一个结果数组ugly中,ugly[0]=1,ugly从l=1开始存值,i,j,k初始都指向位置0,取ugly[i]*2,ugly[j]*3,ugly[k]*5中的最小值,存入ugly[l],然后对应的i,j或k加1。
实现代码
class Solution {
public:
int nthUglyNumber(int n) {
if(n<=0)
return 0;
vector<int> ugly(n);
int i=0,j=0,k=0;
ugly[0]=1;
for(int l=1;l<n;l++){
int tmp=min(ugly[i]*2,min(ugly[j]*3,ugly[k]*5));
if(tmp==ugly[i]*2)i++;
if(tmp==ugly[j]*3)j++;
if(tmp==ugly[k]*5)k++;
ugly[l]=tmp;
}
return ugly[n-1];
}
};