priority_queue:如何创建大顶堆?如何创建自定义类型的堆?

写程序经常会用到priority_queue,大小顶堆,由于用的不熟,今天总结一下留个纪念。

一、基本使用

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

比如这样

priority_queue<int>q;//这是一个int的大顶堆

二、小顶堆

小顶堆的实现有两种方式:

  1. 元素取相反数,比如输入1,2,3,你把取反传进去-1,-2,-3大顶堆取出来数-1,再次取反自然就是你想要的数。
  2. 利用定义实现,根据我们上面的定义 priority_queue<int, vector<int>, greater<int> > q;greater时一个仿函数。

三、自定义类型

必须自己重载 operator< 或者自己写仿函数(有些编译器重载运算符会很坑,还是两个都掌握的比较稳妥)。

  • 重载运算符,推荐采用下面这种方法:
struct cow {
	int s, e;
	cow(int a = 0, int b = 0) { s = a, e = b; }
}c[MAX];

bool operator<(const cow & c1, const cow & c2) {
	return c1.s < c2.s;
}
priority_queue<cow>q;

此时我们将其当作基本类型使用即可。

  • 自己写仿函数,也就是刚才出现的greater<int>这种函数,怎么写呢?
struct cow {
	int s, e;
	cow(int a = 0, int b = 0) { s = a, e = b; }
}c[MAX];
struct cmp {
	bool operator()(cow c1, cow c2) {
		return c1.s < c2.s;
	}
};
priority_queue<cow, vector<cow>, cmp>p;
  • 4
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值