例题5-7 UVA136 丑数 (优先队列的使用)

最新推荐文章于 2020-08-14 11:22:32 发布

原创最新推荐文章于 2020-08-14 11:22:32 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#优先队列

UVA 专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一个使用优先队列和集合数据结构来高效寻找第1500个丑数的C++程序。丑数是指只包含质因数2、3和5的正整数。代码通过不断乘以最小的丑数与2、3和5得到新的丑数，确保了生成的丑数序列的正确性和效率。

#include<iostream>
#include<cstdio>
#include<vector>
#include<queue>
#include<set>
using namespace std;
typedef long long LL;
int a[3]={2,3,5};
int main()
{
    priority_queue<LL,vector<LL>, greater<LL> >q;
    set<LL>s;
    q.push(1);
    s.insert(1);
    for(int i=1;;i++)
    {
        LL ans=q.top();
        q.pop();
        if(i==1500)
        {
            cout<<"The 1500'th ugly number is "<<ans<<"."<<endl;
            break;
        }
        for(int j=0;j<3;j++)
        {
            LL x=ans*a[j];
            if(s.count(x)==0)
            {
                q.push(x);
                s.insert(x);
            }
        }
    }
    return 0;
}