LeetCode之Ugly Number II

最新推荐文章于 2025-04-12 19:07:09 发布

原创最新推荐文章于 2025-04-12 19:07:09 发布 · 324 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Leetcode

Leetcode 专栏收录该内容

168 篇文章

订阅专栏

本文介绍了一种高效算法来寻找第N个丑数。该算法通过维护一个有序数组，并利用2、3、5这三个质因数不断生成新的丑数，确保了每次都能选取最小的丑数加入到数组中。

/*根据提示，保存一个有序的丑数数组，然后通过不断乘以2,3,5，
从中选出最小的数作为下一个丑数。*/
class Solution {
public:
    int nthUglyNumber(int n) {
        vector<int> uglies(1, 1);
        //uglies.reserve(n);
        int i2(0), i3(0), i5(0);
        while(uglies.size() < n){
            int p2 = 2 * uglies[i2], p3 = 3 * uglies[i3], p5 = 5 * uglies[i5];
            int res = min(p2, min(p3, p5));
            if(res == p2) ++i2;
            if(res == p3) ++i3;
            if(res == p5) ++i5;
            uglies.push_back(res);
        }
        return uglies.back();
    }
};