优先队列是队列的一种,不过它可以按照自定义的一种方式(数据的优先级)来对队列中的数据进行动态的排序。每次的push和pop操作,队列都会动态的调整,以达到我们预期的方式来存储。
priority_queue<Type, Container<Type>, Functional>
其中 Type 为数据类型, Container 为保存数据的容器,Functional 为元素比较方式。
Container 必须是用数组实现的容器,比如 vector, deque 但不能用 list,STL里面默认用的是 vector. 比较方式默认用 operator< , 所以如果把后面俩个参数缺省的话,优先队列就是大顶堆,队头元素最大。
// q 是数据类型为 int 型的队列,队头元素[q.top()]最大
priority_queue<int> q;
// p 是数据类型为 int 型的队列,队头元素[p.top()]最小
priority_queue<int, vector<int>, greater<int> > p;
由上可知,如果要用到小顶堆,则要把模板的三个参数都带进去。
greater<> 是 STL里面定义的一个仿函数,对于基本类型可以用这个仿函数声明小顶堆。
对于自定义类型,则必须自己重载“ operator< ”(或者重载“>”)或者自己写仿函数。
struct node
{
friend bool operator< (node n1, node n2)
{
return n1.priority < n2.priority;
// 这里可根据需求改成: return n1.priority > n2.priority;
}
int priority;
int value;
};
思考下下面这段代码:
struct Node {
int weight;
friend bool operator <(Node a, Node b) {
return a.weight > b.weight;
//思考下这里为什么用'>'而不用'<'
}
};
意思是如果a.weight > b.weight成立,那么结构体Node a < Node b成立。由于优先队列是按照从大到小排列,所以结构体b会排列到a之前,然而b.weight是最小的,所以实现了按照weight的从小到大排序。。。。
其实我开始是想这样写的:
struct Node {
int weight;
friend bool operator <(Node a, Node b) {
return a.weight < b.weight;
}
};
// 然后这样使用:(但这种用法会报错)
priority_queue<Node, vector<Node>, greater<Node> > q;
自定义类型重载“operator <”后,声明对象时就可以只带一个模板参数。不能这样声明:priority_queue<Node, vector<Node>, greater<Node> > q
;,如果要这样使用,就要对“operator >”重载。针对前面的题目,代码可改成这样:
struct Node {
int weight;
friend bool operator > (Node a, Node b) {
return a.weight > b.weight; //小顶堆 <的话是大顶堆
}
};
// 然后这样使用(这种用法是可以的):
priority_queue<Node, vector<Node>, greater<Node> > q;
struct Node {
int weight;
friend bool operator < (Node a, Node b) {
return a.weight > b.weight; //小顶堆 < 大顶堆
}
};
// 然后这样使用(这种用法是可以的):
priority_queue<Node> q;
看过这些例子后,可得出如下结论:
对于自定义的类型,重载操作符“<”与只能和声明方法priority_queue<Node> q
;联合使用,而重载操作符“>”则只能对应声明方法priority_queue<Node, vector<Node>, greater<Node> > q