Super Ugly Number

最新推荐文章于 2024-10-02 21:02:12 发布

phlovexz

最新推荐文章于 2024-10-02 21:02:12 发布

阅读量180

点赞数

分类专栏： LeetCode 文章标签： c语言

LeetCode 专栏收录该内容

109 篇文章 0 订阅

订阅专栏

c++

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) {
        vector<int> idx(primes.size(), 0);
        vector<int> ugly(n);
        ugly[0] = 1;
        for (int i = 1; i < n; ++i) {
            ugly[i] = INT_MAX;
            for (int j = 0; j < primes.size(); ++j) {
                ugly[i] = min(ugly[i], primes[j] * ugly[idx[j]]);
            }
            for (int j = 0; j < primes.size(); ++j) {
                while (primes[j] * ugly[idx[j]] <= ugly[i]) {
                    idx[j] ++;
                }
            }
        }
        return ugly.back();
    }
};