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>;