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

h^hh2025-02-01 20:28

当优先级队列里面存的是一个自定义（结构体）类型，我们有两种方式，一个是用内置类型的方式，在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