优先队列的定义
- 在C++的 queue 头文件中定义的优先队列 priority_queue,通常定义方式为
priority_queue<int> s;
- 优先队列还接受其他参数
priority_queue<long long, vector<long long>, greater<long long> > pq;
- 第一个参数 long long 为数据类型
- 第二个参数 vector< long long> 为保存数据的容器,默认为vector< int>
- 第三个参数元素比较函数,优先级,默认为less,less优先级为越大的越先出队列,上面用的 greater 则越小的先出队列
优先队列的使用
empty() 如果队列为空返回 true
pop() 删除队顶元素
push() 加入一个元素
size() 返回优先队列中拥有的元素个数
top() 返回优先队列队顶元素(普通队列中的front()变成了top() )
在默认的优先队列中,优先级高的先出队。在默认的int型中先出队的为较大的数。
使用priority_queue和set集合求丑数
题目描述
丑数是指不能被除了2,3,5以外的其他素数整除的数。把丑数从小到大排列,结果如下:
1,2,3,4,5,6,8,9,10,12,15…
求第1500个丑数
分析
从小到大生成各个丑数,最小的丑数是1,对于任意丑数x,2x、3x和5x都是丑数。这样,可以用一个优先队列保存所有丑数,每次取出最小的丑数,生成3个新的丑数。但需要注意,同一个丑数有多种生成方式,避免重复生成添加丑数。
代码
#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; //set用于存放所有生成过的丑数,可以判断新生成丑数是否已经存在
pq.push(1); //第一个丑数为 1
s.insert(1);
for(int i=1; ;i++){
LL x = pq.top();//取并弹出优先队列队顶元素,第i个丑数
pq.pop();
if(i==1500) {
cout<<"第 1500 个丑数是:"<<x<<endl;
break;
}
for(int j=0;j<3;j++){
LL x2 = x*coeff[j];
if(!s.count(x2)){ //如果新丑数 x2 在 s 集合中不存在
s.insert(x2);
pq.push(x2);
}
}
}
return 0;
}