本篇博文的重点放在元素比较方式上。
1、比较方式采用默认方式
#include<iostream>
#include<queue>
#include<vector>
#include<ctime>
using namespace std;
priority_queue<int>q1;//默认大的优先
priority_queue<int, vector<int>, greater<int>>q2;//默认小的优先
priority_queue<int, vector<int>, less<int>>q3; //默认大的优先
void init()
{
for (int i = 0; i<20; i++)
{
//int x=rand()%100;
q1.push(rand() % 100);
q3.push(rand() % 100);
q2.push(rand() % 100);
}
}
void P(priority_queue<int>q)
{
while (!q.empty())
{
cout << q.top() << " ";
q.pop();
}
cout << endl;
}
void P(priority_queue<int, vector<int>, greater<int> >q)
{
while (!q.empty())
{
cout << q.top() << " ";
q.pop();
}
cout << endl;
}
int main()
{
srand(time(NULL));
init();
P(q1);//(默认)大的优先
P(q2);//小的优先
P(q3);//大的优先
return 0;
}
2、自定义优先级
自定义优先级时,比较函数可以在结构体中定义,此时必须要重载”<”的友元函数,此处有两个重点。
(1)、重载的必须是”<”符合,重载“>”符合会出错。
(2)、必须是友元函数。
#include <iostream>
#include <queue>
#include <ctime>
#include <algorithm>
using namespace std;
struct Node
{
int x, y;
Node() {}
Node( int x,int y):x(x), y(y) {}
//x小的优先,相同时y小的优先
friend bool operator<(Node a, Node b)
{
if( a.x== b.x ) return a.y> b.y;
return a.x> b.x;
}
};
struct cmp
{
//x小的优先,相同时y小的优先
bool operator() ( Node a, Node b )
{
if( a.x== b.x ) return a.y> b.y;
return a.x> b.x;
}
};
int main()
{
priority_queue<Node, vector<Node>, cmp> q;
srand(time(NULL));
for( int i= 0; i< 10; ++i )
q.push( Node( rand()%10, rand()%10 ) );
while( !q.empty() )
{
cout << q.top().x << ' ' << q.top().y << endl;
q.pop();
}
cout<<"------------------------"<<endl;
priority_queue<Node>qq;
for( int i= 0; i< 10; ++i )
qq.push( Node( rand()%10, rand()%10 ) );
while( !qq.empty() )
{
cout << qq.top().x << ' ' << qq.top().y << endl;
qq.pop();
}
return 0;
}