priority_queue的用法

priority_queue本质是一个堆。

1. 头文件是#include<queue>

2. 关于priority_queue中元素的比较

　　模板申明带3个参数：priority_queue<Type, Container, Functional>，其中Type 为数据类型，Container为保存数据的容器，Functional 为元素比较方式。

　　Container必须是用数组实现的容器，比如vector,deque等等，但不能用 list。STL里面默认用的是vector。

2.1 比较方式默认用operator<，所以如果把后面2个参数缺省的话，优先队列就是大顶堆（降序），队头元素最大。特别注意pair的比较函数。

　　以下代码返回一个降序输出：

 1 #include <iostream>
 2 #include <queue>
 3 using namespace std;
 4 int main(){
 5     priority_queue<int> q;
 6     for( int i= 0; i< 10; ++i ) q.push(i);
 7     while( !q.empty() ){
 8         cout<<q.top()<<endl;
 9         q.pop();
10     }
11     return 0;
12 }

　　以下代代码返回pair的比较结果，先按照pair的first元素降序，first元素相等时，再按照second元素降序：

 1 #include<iostream>
 2 #include<vector>
 3 #include<queue>
 4 using namespace std;
 5 int main(){
 6     priority_queue<pair<int,int> > coll;
 7     pair<int,int> a(3,4);
 8     pair<int,int> b(3,5);
 9     pair<int,int> c(4,3);
10     coll.push(c);
11     coll.push(b);
12     coll.push(a);
13     while(!coll.empty())
14     {
15         cout<<coll.top().first<<"\t"<<coll.top().second<<endl;
16         coll.pop();
17     }
18     return 0;
19 }

2.2 如果要用到小顶堆，则一般要把模板的3个参数都带进去。STL里面定义了一个仿函数greater<>，基本类型可以用这个仿函数声明小顶堆。

　　以下代码返回一个升序输出：

 1 #include <iostream>
 2 #include <queue>
 3 using namespace std;
 4 int main(){
 5     priority_queue<int, vector<int>, greater<int> > q;
 6     for( int i= 0; i< 10; ++i ) q.push(10-i);
 7     while( !q.empty() ){
 8         cout << q.top() << endl;
 9         q.pop();
10     }
11     return 0;
12 }

　　以下代代码返回pair的比较结果，先按照pair的first元素升序，first元素相等时，再按照second元素升序：

 1 #include<iostream>
 2 #include<vector>
 3 #include<queue>
 4 using namespace std;
 5 int main(){
 6     priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > > coll;
 7     pair<int,int> a(3,4);
 8     pair<int,int> b(3,5);
 9     pair<int,int> c(4,3);
10     coll.push(c);
11     coll.push(b);
12     coll.push(a);
13     while(!coll.empty())
14     {
15         cout<<coll.top().first<<"\t"<<coll.top().second<<endl;
16         coll.pop();
17     }
18     return 0;
19 }

2.3 对于自定义类型，则必须重载operator<或者重写仿函数。

2.3.1 重载operator<的例子：返回true时，说明左边形参的优先级低于右边形参

 1 #include <iostream>
 2 #include <queue>
 3 using namespace std;
 4 struct Node{
 5     int x, y;
 6     Node(int a=0, int b=0):
 7         x(a),y(b){}
 8 };
 9 bool operator<(Node a, Node b){//返回true时，说明a的优先级低于b
10     //x值较大的Node优先级低（x小的Node排在队前）
11     //x相等时，y大的优先级低（y小的Node排在队前）
12     if( a.x== b.x ) return a.y> b.y;
13     return a.x> b.x;
14 }
15 int main(){
16     priority_queue<Node> q;
17     for( int i= 0; i< 10; ++i )
18     q.push( Node( rand(), rand() ) );
19     while( !q.empty() ){
20         cout << q.top().x << ' ' << q.top().y << endl;
21         q.pop();
22     }
23     return 0;
24 }

自定义类型重载operator<后，声明对象时就可以只带一个模板参数。

但此时不能像基本类型这样声明priority_queue<Node,vector<Node>,greater<Node> >，原因是greater<Node>没有定义，如果想用这种方法定义则可以重载operator >。

例子：返回的是小顶堆。但不怎么用，习惯是重载operator<。

 1 #include <iostream>
 2 #include <queue>
 3 using namespace std;
 4 struct Node{
 5     int x, y;
 6     Node( int a= 0, int b= 0 ):
 7         x(a), y(b) {}
 8 };
 9 bool operator>( Node a, Node b ){//返回true，a的优先级大于b
10     //x大的排在队前部；x相同时，y大的排在队前部
11     if( a.x== b.x ) return a.y> b.y;
12     return a.x> b.x;
13 }
14 int main(){
15     priority_queue<Node,vector<Node>,greater<Node> > q;
16     for( int i= 0; i< 10; ++i )
17     q.push( Node( rand(), rand() ) );
18     while( !q.empty() ){
19         cout << q.top().x << ' ' << q.top().y << endl;
20         q.pop();
21     }
22     return 0;
23 }

2.3.2 重写仿函数的例子（返回值排序与2.3.1相同，都是小顶堆。先按x升序，x相等时，再按y升序）：

 1 #include <iostream>
 2 #include <queue>
 3 using namespace std;
 4 struct Node{
 5     int x, y;
 6     Node( int a= 0, int b= 0 ):
 7         x(a), y(b) {}
 8 };
 9 struct cmp{
10     bool operator() ( Node a, Node b ){//默认是less函数
11         //返回true时，a的优先级低于b的优先级（a排在b的后面）
12         if( a.x== b.x ) return a.y> b.y;
13         return a.x> b.x; }
14 };
15 int main(){
16     priority_queue<Node, vector<Node>, cmp> q;
17     for( int i= 0; i< 10; ++i )
18     q.push( Node( rand(), rand() ) );
19     while( !q.empty() ){
20         cout << q.top().x << ' ' << q.top().y << endl;
21         q.pop();
22     }
23     return 0;
24 }

参考：http://www.cnblogs.com/flyoung2008/articles/2136485.html

转载于:https://www.cnblogs.com/Deribs4/p/5657746.html

priority_queue的用法相关推荐

priority_queue 用法总结
今天在写堆和哈夫曼树的ACM题的时候,接触到priority_queue的用法,由于比较函数的难些,请教过队内的红薯和杨大牛后才稍微弄明白些,下面总结如下,首先我是用手写的堆来过题的,其实和照黑书指导 ...
c++ stl容器vector删除(erase),遍历等基本用法介绍及头文件
Vectors 包含着一系列连续存储的元素,其行为和数组类似.访问Vector中的任意元素或从末尾添加元素都可以在常量级时间复杂度内完成,而查找特定值的元素所处的位置或是在Vector中插入元素则是线 ...
理解至上：二叉堆与优先队列详细用法
文章目录二叉堆概述插入代码访问代码完整代码优先队列:priority_queue 基本用法小根堆的声明: 结构体注意 Thanks for reading! 二叉堆概述为什么不 ...
C++ queue和priority_queue
queue 和 priority_queue 都是容器适配器,要使用它们,必须包含头文件 . queue queue 就是"队列".队列是先进先出的,和排队类似.队头的访问和删除操 ...
STL-priority_queue用法（重点: 升序，小根堆）
文章出处:http://www.cnblogs.com/mfryf/archive/2012/09/05/2671883.html STL-priority_queue用法(重点: 升序,小根堆) 昨 ...
C++_STL——stack
C++_STL--stack template <class T, class Container = deque<T> > class stack; 堆栈后进先出堆栈堆栈 ...
今日头条的面试题（部分）
1.给定一个有序数组,输出平方后消重结果中数字的个数,每次删除一个最大的数比如: -2, -2, -1,0,1消重后有3个数: 0,0,1,2,3,4,5,5,6消重后有7个数 O(n)复杂度解法: ...
nyoj55懒省事的小明
这个题没啥特别的,思想特别简单,就是我们学数据结构的时候,那里面的哈夫曼树的思想,从一组数里面选两个最小的相加,将这两个数从这组数中删除,再将这两个数的和放进去,再从这组数中选两个最小的数以此类推, ...
C++ priority_queue用法
列举了四种用法,分别是默认降序.内置的greater升序.重载<运算符.仿函数. #include <cstdio> #include <cstring> #includ ...

priority_queue的用法

priority_queue的用法相关推荐

最新文章

热门文章