[转]: STL priority_queue 优先队列

 

刚开始学习算法不久,一些常用的算法工具还没有掌握,真是丢人!

前一段时间用到优先级队列时,都是自己手动通过最大堆或者最小堆来写一个,容易出错且耗时。接触到STL后,开始用map和set模拟一个优先级队列,但是总有一些小问题出现,发现STL功能强大,但我却几乎不懂。

今天终于决定使用STL提供的priority_queue,发现还挺好用,虽然很多人都称他效率不够高,但是使用起来很方便。下面就总结一下它的一般用法:

模板原型:

priority_queue<T,Sequence,Compare>

T:存放容器的元素类型

Sequence:实现优先级队列的底层容器,默认是vector<T>

Compare:用于实现优先级的比较函数,默认是functional中的less<T>

常用的操作如下:

empty()  如果优先队列为空,则返回真 
pop()  删除第一个元素 
push()  加入一个元素 
size()  返回优先队列中拥有的元素的个数 
top()  返回优先队列中有最高优先级的元素


但是在使用时必须注意:priority_queue放置元素时,不会判断元素是否重复。(因为在模板的第二个参数时顺序容器,不能保证元素的唯一性)

此外可以替代默认的Compare函数。下面用一个小例子说明一下:

 

  1. #include <iostream>  
  2. #include <queue>  
  3. #include <vector>  
  4. #include <functional>  
  5. using namespace std;  
  6. struct Node  
  7. {  
  8.     int f;  
  9.     bool operator<(const Node& node)const  
  10.     {  
  11.         return f>node.f;  
  12.     }  
  13. };  
  14. class cmp   
  15. {  
  16.     public:  
  17.     bool operator()( const Node & n1, const Node & n2) const  
  18.     {  
  19.         return n1<n2;  
  20.     }  
  21. };  
  22. int main()  
  23. {  
  24.     priority_queue< Node,vector<Node>,cmp > q;  
  25.     Node n1;  
  26.     n1.f = 5;  
  27.     Node n2;  
  28.     n2.f = 4;  
  29.     Node n3;  
  30.     n3.f = 2;  
  31.     Node n4;  
  32.     n4.f = 10;  
  33.     q.push(n1);  
  34.     q.push(n2);  
  35.     q.push(n3);  
  36.     q.push(n4);  
  37.     while(!q.empty())  
  38.     {  
  39.         cout<< q.top().f<<endl;  
  40.         q.pop();  
  41.     }  
  42. }  

 

在自己实现Compare函数(函数对象)时,函数对象是中的参数如果是const类型,那么队列存放的元素(比如Node)的比较函数也一定要实现为const。因为函数对象在执行时通过其参数调用Node的比较函数。

 

另一简明说明 :

在优先队列中,优先级高的元素先出队列。
标准库默认使用元素类型的<操作符来确定它们之间的优先级关系。
优先队列的第一种用法,也是最常用的用法:

priority_queue < int >  qi;

通过<操作符可知在整数中元素大的优先级高。
故示例1中输出结果为:9 6 5 3 2

第二种方法:
在示例1中,如果我们要把元素从小到大输出怎么办呢?
这时我们可以传入一个比较函数,使用functional.h函数对象作为比较函数。

priority_queue < int , vector < int > , greater < int >   > qi2;

其中
第二个参数为容器类型。
第二个参数为比较函数。
故示例2中输出结果为:2 3 5 6 9

第三种方法:
自定义优先级。

struct node
{
     friend  bool operator <  (node n1, node n2)
    {
         return  n1.priority  <  n2.priority;
    }
     int  priority;
     int  value;
};

在该结构中,value为值,priority为优先级。
通过自定义operator<操作符来比较元素中的优先级。
在示例3中输出结果为:
优先级  值
9          5
8          2
6          1
2          3
1          4
但如果结构定义如下:

struct node
{
     friend  bool operator >  (node n1, node n2)
    {
         return  n1.priority  >  n2.priority;
    }
     int  priority;
     int  value;
};

则会编译不过(G++编译器)
因为标准库默认使用元素类型的<操作符来确定它们之间的优先级关系。
而且自定义类型的<操作符与>操作符并无直接联系,故会编译不过。

//代码清单

 

#include<iostream>
#include<functional>
#include<queue>
using namespace std;
struct node
{
    friend bool operator< (node n1, node n2)
    {
        return n1.priority < n2.priority;
    }
    int priority;
    int value;
};
int main()
{
    const int len = 5;
    int i;
    int a[len] = {3,5,9,6,2};
    //示例1
    priority_queue<int> qi;
    for(i = 0; i < len; i++)
        qi.push(a[i]);
    for(i = 0; i < len; i++)
    {
        cout<<qi.top()<<" ";
        qi.pop();
    }
    cout<<endl;
    //示例2
    priority_queue<int, vector<int>, greater<int> >qi2;
    for(i = 0; i < len; i++)
        qi2.push(a[i]);
    for(i = 0; i < len; i++)
    {
        cout<<qi2.top()<<" ";
        qi2.pop();
    }
    cout<<endl;
    //示例3
    priority_queue<node> qn;
    node b[len];
    b[0].priority = 6; b[0].value = 1; 
    b[1].priority = 9; b[1].value = 5; 
    b[2].priority = 2; b[2].value = 3; 
    b[3].priority = 8; b[3].value = 2; 
    b[4].priority = 1; b[4].value = 4; 

    for(i = 0; i < len; i++)
        qn.push(b[i]);
    cout<<"优先级"<<'/t'<<"值"<<endl;
    for(i = 0; i < len; i++)
    {
        cout<<qn.top().priority<<'/t'<<qn.top().value<<endl;
        qn.pop();
    }
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值