STL中的优先队列也定义在头文件中,用 priority_queuep 来声明。这个 p 是一个"越小的整数优先级越低的优先队列“。由于出队元素并不是最先进队列的元素,出队的方法由queue 中的front()变为了top()。
自定义类型也可以组成优先队列,但必须为每一个元素定义一个优先级。这个优先级并不用一个确切的数字,只需要可以比较大小就行。例如,要实现一个”个位数大的优先级反而小“的优先队列,可以定义一个结构体cmp,重载”()“运算符,使其”看上去“像一个”函数“,然后用”priority_queue<int,vector,cmp>p“ 的方式定义。,下面是cmp的定义:
struct cmp{
bool operator() (const int a,const int b) const{
return a % 10 > b % 10;
}
};
对于一些常见的优先队列,STL提供了更为简单的定义方法,例如越小的整数优先级越大的优先级队列可以写成”priority_queue<int,vector,greater >p“,注意最后两个”>“符号不要写在一起,否则会被很多编译器误认为”>>“运算符。
Description
编写一个程序,找出第 n 个丑数。
丑数就是只包含质因数 2, 3, 5 的正整数。
注:1是特殊的丑数。
示例:
输入: n = 10
输出: 12
解释: 1, 2, 3, 4, 5, 6, 8, 9, 10, 12 是前 10 个丑数。
Input
输入一个数n (1 ≤ n ≤ 1500)。
Output
输出第n个丑数。
Sample Input
10
Sample Output
12
分析:从小到大生成各个丑数,最小的为1,对于任意抽数x,2x,3x,5x,也是丑数,可以用一个优先级队列保存所有已经生成的丑数,每次取出最小的丑数,生成新的3个丑数,对每个丑数标记是否之前生成过。
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
priority_queue<ll, vector<ll> ,greater<ll> >p;
set<ll> s;
int main(){
int n,coff[3]={2,3,5};
cin>>n;
p.push(1);
s.insert(1);
for (int i = 1; ; i++){
ll x = p.top();
p.pop();
if(i == n){
cout<<x<<endl;
break;
}
for(int j = 0 ;j< 3;j++){
ll x2 = x * coff[j];
if(!s.count(x2)){
s.insert(x2);
p.push(x2);
}
}
}
return 0;
}
end.