STL 自定义比较函数

最新推荐文章于 2024-06-03 23:15:48 发布

Zarxdy34

最新推荐文章于 2024-06-03 23:15:48 发布

阅读量811

点赞数

分类专栏： STL 文章标签：自定义比较函数 STL

本文链接：https://blog.csdn.net/Zarxdy34/article/details/45268465

版权

本文通过举例说明如何使用STL中的priority_queue创建小根堆。通过将默认的greater<int>比较函数替换为less<int>，实现了元素类型的反转，从而达到小根堆的效果。同时探讨了在需要自定义元素类型或者维护多个不同性质堆的情况下，如何自定义less和greater比较函数。

摘要由CSDN通过智能技术生成

以priority_queue为例。

如果我要写一个元素类型为整型的大根堆，那么直接用STL的优先队列就可以了。

#include <cstdio>
#include <queue>
using namespace std;

priority_queue <int> Q;

int main()
{
	Q.push(1);
	printf("%d\n",Q.top());
	Q.pop();
	return 0;
}

但有些时候我们需要使用小根堆，直接使用STL还是可以实现的。

#include <cstdio>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;

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

int main()
{
	Q.push(1);
	Q.push(2);
	printf("%d\n",Q.top());
	Q.pop();
	return 0;
}

输出是1，符合小根堆。简单地理解一下，greater<int>是int的比较函数，这个位置原本放的应该是“小于”的比较函数，但是这时我放了一个“大于”的比较函数上去，比较函数恰好反了过来，那么原来大根堆就变成小根堆了。

如果要自定义元素类型？