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
相关推荐
梁辰兴1 小时前
数据结构:排序
数据结构·算法·排序算法·c·插入排序·排序·交换排序
Murphy_lx1 小时前
Lambda表达式
开发语言·c++
yangpipi-1 小时前
C++并发编程-23. 线程间切分任务的方法
开发语言·c++
野犬寒鸦1 小时前
力扣hot100:搜索二维矩阵 II(常见误区与高效解法详解)(240)
java·数据结构·算法·leetcode·面试
菜鸟得菜1 小时前
leecode kadane算法 解决数组中子数组的最大和,以及环形数组连续子数组的最大和问题
数据结构·算法·leetcode
爬虫程序猿2 小时前
利用 Java 爬虫获取淘宝商品 SKU 详细信息实战指南
java·开发语言·爬虫
F2E_Zhangmo2 小时前
基于cornerstone3D的dicom影像浏览器 第五章 在Displayer四个角落显示信息
开发语言·前端·javascript
楼田莉子2 小时前
C++算法专题学习——分治
数据结构·c++·学习·算法·leetcode·排序算法