这是lintcode上的一道题:原题地址
丑数:是素因子只有2,3,5的数,有的人说这句话不好理解,其实意思就是一个数有很多因子,这些因子中如果有素数,只能是2,3,5,举个例子,14有因子7,7是14的素因子,不在2,3,5的范围内,故14不是丑数。
思路就不说了,网上都是说丑数是之前的丑数的2,3,5倍啊,只要排序就行了。
我在九章那边看到了一种比较简洁的写法。用一个数组,然后就依次找这些数的2,3,5倍。比如第二个数的3倍小了(已经在丑数数组中了),就找第三个数的3倍;如果第四个数的5倍小了,就找第五个数的五倍。相当于2,3,5在分三路分别遍历。里面我觉得比较巧妙的是寻找下一个数是通过移动指针实现的。
以下是代码,我发现原来的代码,解引用多余还有点误导人,所以就改了一下。
int nthUglyNumber(int n) {
int *uglys = new int[n];
uglys[0] = 1;
int next = 1;
int *p2 = uglys;
int *p3 = uglys;
int *p5 = uglys;
while(next < n) {
uglys[next] = min(min(*p2*2,*p3*3),*p5*5);
while(*p2 * 2 <= uglys[next])
p2++;
while(*p3 * 3 <= uglys[next])
p3++;
while(*p5 * 5 <= uglys[next])
p5++;
next++;
}
int result = uglys[n - 1];
delete [] uglys;
return result;
}