UVA136 Ugly Numbers(优先队列应用)

最新推荐文章于 2020-12-10 20:34:57 发布

原创最新推荐文章于 2020-12-10 20:34:57 发布 · 618 阅读

0 ·

CC 4.0 BY-SA版权

UVA 专栏收录该内容

232 篇文章

订阅专栏

本文深入探讨了使用优先队列技术解决生成特定数列的问题，以生成第1500个丑数为例，展示了算法的实现过程及优化思路。

优先队列：“急诊病人插队”

priority_queue<int> pq声明优先队列

"越小的整数优先级越低的队列"

push()入队

pop()出队

top()取队首元素

这个题像了一会，纠结在为什么i == 1500

时停，原因就在于这个大循环每次会把最小的那个整数提出来，制造3个新的丑数，然后出队。这样每一循环，按小到大顺序出队一个丑数，直到1500个

#include<iostream>
#include<vector>
#include<queue>
#include<set>
using namespace std;
typedef long long LL;
const int coeff[3] = {2, 3, 5};//coeff数组存放最基础的丑数2,3,4

int main()
{
    priority_queue<LL, vector<LL>, greater<LL> > pq;//定义优先队列
    set<LL> s;//集合s作用是
    pq.push(1);//把1入队
    s.insert(1);//集合中插入1
    for(int i  = 1; ; i++) {
        LL x = pq.top(); pq.pop();//把优先队列中最小的数放到x里， 把x出队
        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];//用x生成3个丑数
            if(!s.count(x2)) { s.insert(x2); pq.push(x2); }//如果这个数没出现过，就入队，并且插入到集合中
        }
    }
    return 0;
}