领扣刷题日志(丑数2)

问题描述:设计一个算法,找出只含素因子235 的第 n 小的数。

解题思路: 按顺序存储前n个丑数,返回第n个数。

答案:

int nthUglyNumber(int n) {
        // write your code here
        vector 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)

你可能感兴趣的:(Lintcode)