B - Ugly Numbers

最新推荐文章于 2020-03-15 18:24:22 发布

原创最新推荐文章于 2020-03-15 18:24:22 发布 · 385 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#优先队列 #栈 #set

紫书例题专栏收录该内容

3 篇文章

订阅专栏

B - Ugly Numbers

题意

丑数是指不能被2，3，5以外的其他素数整除的数。把丑数从小到大排列起来，结果如下：
1,2,3,4,5,6,8,9,10,12,15……
求第1500个丑数

输入

没有输入

输出

The 1500'th ugly number is <number>.

#include <iostream>
#include <vector>
#include <queue>
#include <set>
using namespace std;
typedef long long LL;
const int coeff[3] = {2,3,5};

int main()
{
    priority_queue<LL,vector<LL>,greater<LL> > pq;
    set<LL> s;
    pq.push(1);
    s.insert(1);
    for(int i = 1;;i++)
    {
        LL x = pq.top();
        pq.pop();
        if(i == 1500)
        {
            cout << "The 1500'th ugly number is "<< x <<".\n";
            break;
        }
        for(int j = 0;j < 3;j++)
        {
            LL x2 = x * coeff[j];
            if(!s.count(x2))
            {
                s.insert(x2);
                pq.push(x2);
            }
        }
    }
    return 0;
}