C++ 优先队列用法自定义Compare函数

C++头文件 #include <queue>

template<typename _Tp,
             typename _Sequence = vector<_Tp>,
             typename _Compare  = less<typename _Sequence::value_type> >

第一个参数 _Tp: 指定存储的类型名称;

第二个参数 _Sequence: 指定存储的数据结构,该结果必须支持随机存取迭代器;

第三个参数 _Compare : 比较函数,对于自定义类型有两种方法实现大小顶堆,第一个是重载操作符,第二个是写一个结构实现比较。

各个数据类型算法讲解如下:

1.  对于一般的基本数据类型,比如 int,double等。

1).    默认是大顶堆,测试代码如下

priority_queue<int,vector<int>> pqueue_int;
    for(int i = 0 ; i < 10 ; ++i){
        pqueue_int.push(rand()%100);
    }
    while(!(pqueue_int.empty()))
    {
        cout<<pqueue_int.top()<<"\t";
        pqueue_int.pop();
    }
    cout<<endl;

其中省略了比较泛型函数less<int>

输出结果如下:


2).   小顶堆实现如下:

priority_queue<int,vector<int>,greater<int>> pqueue_int;
    for(int i = 0 ; i < 10 ; ++i){
        pqueue_int.push(rand()%100);
    }
    while(!(pqueue_int.empty()))
    {
        cout<<pqueue_int.top()<<"\t";
        pqueue_int.pop();
    }
    cout<<endl;

其中加上了存储的数据结构,和比较函数大小。

输出结果如下:


2.  对于自定义类型,必须实现比较函数。

自定义类型如下:

struct Edge{
    int start,end;
    double weight;
    Edge(int start,int end,int weight):start(start),end(end),weight(weight){};
};
typedef struct Edge Edge;


bool operator > (Edge a,Edge b)
{
    return a.weight > b.weight;
}

int main()
{
    priority_queue<Edge,vector<Edge>,greater<Edge>> pqueue_Edge;
    pqueue_Edge.push(Edge(0,1,3));
    pqueue_Edge.push(Edge(0,2,4));
    pqueue_Edge.push(Edge(0,3,8));
    pqueue_Edge.push(Edge(0,4,5));


    while(!pqueue_Edge.empty()){
        Edge temp = pqueue_Edge.top();
        pqueue_Edge.pop();
        cout<<temp.start<<"\t"<<temp.end<<"\t"<<temp.weight<<endl;
    }
}

运行结果:


展开阅读全文

没有更多推荐了,返回首页