priority_queue用法

在优先队列中，优先级高的元素先出队列。
 

先写一个用 STL 里面堆算法实现的与真正的STL里面的 priority_queue 用法相
似的 priority_queue， 以加深对 priority_queue 的理解

push_heap()：将容器中的最后一个元素加入堆中

pop_head()：将堆中最大的(或者自定义比较函数，默认为<)元素推到容器首

 
 
#include <iostream>  
#include <algorithm>  
#include <vector>  
 
using namespace std;  
 
class priority_queue  
{  
    private:  
        vector<int> data;  
          
    public:  
        void push( int t ){   
            data.push_back(t);   
            push_heap( data.begin(), data.end());   
        }  
          
        void pop(){  
            pop_heap( data.begin(), data.end() );  
            data.pop_back();  
        }  
          
        int top()    { return data.front(); }  
        int size()   { return data.size();  }  
        bool empty() { return data.empty(); }  
};  
 
int main()  
{  
    priority_queue  test;  
    test.push( 3 );  
    test.push( 5 );  
    test.push( 2 );  
    test.push( 4 );  
      
    while( !test.empty() ){  
        cout << test.top() << endl;  
        test.pop(); }  
          
    return 0;  
}  

 运行结果：

 

STL里面的 priority_queue 写法与此相似，只是增加了模板及相关的迭代器什么的。

priority_queue 对于基本类型的使用方法相对简单。
他的模板声明带有三个参数，priority_queue<Type, Container, Functional>
Type 为数据类型， Container 为保存数据的容器，Functional 为元素比较方式。
Container 必须是用数组实现的容器，比如 vector, deque 但不能用 list.
STL里面默认用的是 vector. 比较方式默认用 operator< , 所以如果你把后面俩个
参数缺省的话，优先队列就是大顶堆，队头元素最大。
(1)使用默认参数

 
 
#include <iostream>  
#include <queue>  
using namespace std;  
 
int main(){  
    priority_queue<int> q;  
      
    for( int i= 0; i< 10; ++i )  
        q.push(i%4);  
 
    while( !q.empty() ){  
        cout << q.top() << endl;  
        q.pop();  
    }  
    return 0;  
} 

(2)自定义比较函数(可以使用STL里的函数greater<int>等，或者重载<，这样就不需要用三个参数，或者自定义比较函数)

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

 

bool operator<  (int a,int b){return a < b;}

priority_queue<int>

 

bool cmp(int a,int b) { return a < b;}

priority_queue<int,vector<int>,cmp>;

 

本文出自 “牛哥的博客” 博客，请务必保留此出处http://nxlhero.blog.51cto.com/962631/699349