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

最新推荐文章于 2025-10-31 16:48:40 发布

原创最新推荐文章于 2025-10-31 16:48:40 发布 · 8.2k 阅读

5 ·

CC 4.0 BY-SA版权

文章标签：

#c语言 #C++ #STL

C/C++ 同时被 2 个专栏收录

2 篇文章

订阅专栏

C/C++_STL

1 篇文章

订阅专栏

本文介绍了C++中优先队列的使用，重点在于如何为自定义类型提供比较函数。通过示例展示了如何创建大顶堆和小顶堆，并给出了自定义类型Edge的代码，演示了如何利用greater<Edge>实现小顶堆。

部署运行你感兴趣的模型镜像

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

运行结果：