priority_queue的创建_结构体类型(重载小于运算符)c++

当优先级队列里面存的是一个自定义(结构体)类型,我们有两种方式,一个是用内置类型的方式,在priority_queue<>里写三个参数,比如int, vector<int>, less<int>,把int改成结构体的名字就可以了,但此时就不能用less了,因为less默认是对内置类型使用的,如果less<>里面写自定义类型,还需要写小于号重载运算符的,不然会报错,如果你不想用less,你自己在重新写一个比较方法才行,greater同理,所以如果是结构体的话,我们按照less写法有点麻烦

用第二种写法,在结构体中直接重载小于号运算符,直接把创建出来的结构体,放到优先级队列里,就可以创建出相应的大小根堆

cpp 复制代码
#include <iostream>
#include <queue>
using namespace std;

struct node {
	int a, b, c;

	//重载<运算符
	//按照b为基准,创建大根堆
	//bool operator<(const node& x) const //最后面的const表示这个函数是一个常量成员函数
	//{
	//	return b < x.b;
	//}

	//小根堆,把<改成>就可以了
	bool operator<(const node& x) const
	{
		return b > x.b;
	}
};

void test()
{
	//如果选择重载小于运算符的话,创建堆的时候直接把node放进来就可以了
	priority_queue<node> heap;

	for (int i = 1; i <= 5; ++i)
	{
		heap.push({ i + 5, i + 1, i + 2 });
	}

	while (heap.size())
	{
		node t = heap.top(); heap.pop();
		cout << t.a << " " << t.b << " " << t.c << endl;
	}
}

int main()
{
	test();

	return 0;
}
输出:
6 2 3
7 3 4
8 4 5
9 5 6
10 6 7

其实我们掌握了重载小于运算符,也会重载大于运算符,这些就够了,感兴趣也可以看看第一种less写法,也是写个小于重载,greater在实现个大于重载运算符

cpp 复制代码
#include <iostream>
#include <queue>
using namespace std;

struct node {
	int a, b, c;

	//less 大根堆
	bool operator<(const node& x) const { return b < x.b; }

	//greater 小根堆
	bool operator>(const node& x) const { return b > x.b; }
};


void test2()
{
	//小根堆
	priority_queue<node, vector<node>, greater<node>> heap;

	for (int i = 1; i <= 5; ++i)
	{
		heap.push({ i + 5, i + 1, i + 2 });
	}

	while (heap.size())
	{
		node t = heap.top(); heap.pop();
		cout << t.a << " " << t.b << " " << t.c << endl;
	}
}

int main()
{
	test2();

	return 0;
}

输出:
6 2 3
7 3 4
8 4 5
9 5 6
10 6 7
相关推荐
csbysj202040 分钟前
Perl 格式化输出
开发语言
tao3556671 小时前
【Python刷力扣hot100】42. Trapping Rain Water
开发语言·python·leetcode
Miraitowa_cheems1 小时前
LeetCode算法日记 - Day 88: 环绕字符串中唯一的子字符串
java·数据结构·算法·leetcode·深度优先·动态规划
消失的旧时光-19432 小时前
Kotlin 协程最佳实践:用 CoroutineScope + SupervisorJob 替代 Timer,实现优雅周期任务调度
android·开发语言·kotlin
错把套路当深情2 小时前
Kotlin保留小数位的三种方法
开发语言·python·kotlin
B站_计算机毕业设计之家2 小时前
python电商商品评论数据分析可视化系统 爬虫 数据采集 Flask框架 NLP情感分析 LDA主题分析 Bayes评论分类(源码) ✅
大数据·hadoop·爬虫·python·算法·数据分析·1024程序员节
小白菜又菜2 小时前
Leetcode 1518. Water Bottles
算法·leetcode·职场和发展
长存祈月心3 小时前
Rust Option 与 Result深度解析
算法
赵谨言3 小时前
基于Python Web的大数据系统监控平台的设计与实现
大数据·开发语言·经验分享·python
专注前端30年3 小时前
Vue2 中 v-if 与 v-show 深度对比及实战指南
开发语言·前端·vue