AcWing 136. UVA (priority_queue)
原题链接
简单
作者:
史一帆
,
2021-04-19 11:31:53
,
所有人可见
,
阅读 328
//升序队列,小顶堆
priority_queue <int,vector<int>,greater<int> > q;
//降序队列,大顶堆
priority_queue <int,vector<int>,less<int> >q;
//greater和less是std实现的两个仿函数(就是使一个类的使用看上去像一个函数。其实现就是类中实现一个operator(),这个类就有了类似函数的行为,就是一个仿函数类了)
代码实现
#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 << '.' << endl;
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;
}