313. Super Ugly Number

原创于 2017-03-07 21:01:19 发布 · 217 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

429 篇文章

订阅专栏

本文介绍了一种使用大顶堆实现寻找第n个超级丑数的方法。通过定义一个大顶堆并遵循特定的算法步骤，可以有效地找到超级丑数序列中的指定项。此方法避免了重复计算，并确保了每次迭代都能获取到当前最小的丑数。

简单大顶堆。

class Solution {
public:
    int nthSuperUglyNumber(int n, vector<int>& primes) {
        std::priority_queue<int, std::vector<int>, std::greater<int> > nums;
        nums.push(1);

        while(--n)
        {
            //cout<<nums.top()<<endl;
            int topNum=nums.top();
            nums.pop();
            while((!nums.empty())&&nums.top()==topNum)
                nums.pop();


            for(int i=0;i<primes.size();i++)
                if(topNum<=INT_MAX/primes[i])
                    nums.push(topNum*primes[i]);
        }

        return nums.top();

    }
};