【C++优先队列 priority_queue】求丑数

优先队列的定义

• 在C++的 queue 头文件中定义的优先队列 priority_queue，通常定义方式为

priority_queue<int> s；

• 优先队列还接受其他参数

priority_queue<long long, vector<long long>, greater<long long> > pq;

1. 第一个参数 long long 为数据类型
2. 第二个参数 vector< long long> 为保存数据的容器，默认为vector< int>
3. 第三个参数元素比较函数，优先级，默认为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;
}