STL的优先队列定义在<queue>中,用“priority_queue<int>pq”来声明。这个pq是越小的整数优先级越低的队列,由于出队元素不是先入队的元素,所以front()变为了top()
越小的整数优先级越大"priority_queue<int,vector<int>,greater<int> >pq",注意最后的两个>需要用空格隔开,否则会被认为是">>"运算符
丑数:设一个丑数为x,那么2x,3x,5x都为丑数,所以从1开始,逐步推出丑数放在集合和优先队列中,每次推导时,取出队列中最小的值,然后删去,依次类推
#include<cstdio> #include<iostream> #include<vector> #include<queue> #include<set> using namespace std; const int a[3] = { 2,3,5 }; int main(void) { typedef long long LL; priority_queue<LL, vector<LL>, greater<LL> > pq; set<LL> stock; stock.insert(1); pq.push(1); for (int i = 1;; i++) { LL t = pq.top(); pq.pop(); if (i == 1500) { cout << "The 1500'th ugly number is " << t << "." << endl ; break; } else { for (int j = 0; j < 3; j++) { LL x = t * a[j]; if (!stock.count(x)) { stock.insert(x); pq.push(x); } } } } return 0; }