(Lint Code 4)用C++实现查找丑数

这是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;
}

 

你可能感兴趣的:(practice)