问题描述:设计一个算法,找出只含素因子2
,3
,5
的第 n 小的数。
解题思路: 按顺序存储前n个丑数,返回第n个数。
答案:
int nthUglyNumber(int n) {
// write your code here
vector<int> result;
result.push_back(1);
int p1 = 0; int p2 = 0; int p3 = 0;
for(int i = 0; i < n; i++){
int biggest = result[i];
while(result[p1]*2 <= biggest) p1++;
while(result[p2]*3 <= biggest) p2++;
while(result[p3]*5 <= biggest) p3++;
int temp = min(result[p1]*2, min(result[p2]*3, result[p3]*5));
result.push_back(temp);
}
return result[n-1];
}
算法解释:
一个丑数乘以2,3或是5,得到的一定是丑数。
p1,p2,p3的作用:在第i次循环刚刚开始时,找到result中已有的且乘以2(3或5)就会大于biggest的丑数。它们存在的意义是确保即将存储的丑数是大于biggest的最小丑数,进而确保result中的n个丑数为最小的n个连续的丑数。
答案思路来源于:九章算法(http://www.jiuzhang.com/?source=code)