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
相关推荐
lly20240624 分钟前
PostgreSQL 表达式
开发语言
LXMXHJ25 分钟前
php开发
开发语言·php
程序员大雄学编程1 小时前
「深度学习笔记4」深度学习优化算法完全指南:从梯度下降到Adam的实战详解
笔记·深度学习·算法·机器学习
mjhcsp1 小时前
MATLAB 疑难问题诊疗:从常见报错到深度优化的全流程指南
开发语言·matlab
Lynnxiaowen1 小时前
今天我们开始学习python语句和模块
linux·运维·开发语言·python·学习
逐步前行1 小时前
C标准库--浮点<float.h>
c语言·开发语言
小O的算法实验室1 小时前
2022年ASOC SCI2区TOP,基于竞争与合作策略的金字塔粒子群算法PPSO,深度解析+性能实测,深度解析+性能实测
算法·论文复现·智能算法·智能算法改进
深耕AI2 小时前
MFC + OpenCV 图像预览显示不全中断问题解决:GDI行填充详解
c++·opencv·mfc
武帝为此2 小时前
【B树与B+树详解】
数据结构·b树
zoyation2 小时前
多线程简介和在JAVA中应用
java·开发语言