C++中优先队列的基本使用(基本类型&自定义类型)

   日期:2020-05-11     浏览:261    评论:0    
核心提示:给出优先队列中对于基本类型和自定义类型的使用c/c++

优先队列(priority_queue)定义在头文件< queue > 中,可以用来实现二叉堆(大顶堆、小顶堆),主要使用包括对于基本类型和自定义类型的排序,顶部元素(优先级最高/最低)访问,出队、入队等操作。

  1. 对于基本类型,可以修改默认排序规则,将大顶堆该为小顶堆。
  2. 对于自定义数据类型,可以创建操作符重载函数(重载<符号),或者自定义比较函数(重载()符号),用于自定义类型的比较操作。

接下来给出优先队列的基本使用:

test_01() : 使用int类型
test_02(): 使用string类型
test_03(): 使用自定义类型struct student

#include <iostream>
#include <vector>
#include <queue>
#include <algorithm>

using namespace std;

// 优先队列:基本数据类型的使用 - int
void test_01()
{
	// 默认为大顶堆
	priority_queue<int> p1;
	p1.push(10);
	p1.push(-10);
	p1.push(100);

	cout << "p1 list 大顶堆(int): ";
	while (p1.size())
	{
		cout << p1.top() << " ";
		p1.pop();
	}
	cout << endl;

	// 改为小顶堆实现
	priority_queue<int, vector<int>, greater<int>> p2;
	p2.push(10);
	p2.push(-10);
	p2.push(100);

	cout << "p2 list 小顶堆(int): ";
	while (p2.size())
	{
		cout << p2.top() << " ";
		p2.pop();
	}
	cout << endl;
}

// 优先队列:基本数据类型的使用 - string
void test_02()
{
	// 默认为大顶堆: C B A
	priority_queue<string> p3;
	p3.push("A");
	p3.push("B");
	p3.push("C");

	cout << "p3 list 大顶堆(string): ";
	while (p3.size())
	{
		cout << p3.top() << " ";
		p3.pop();
	}
	cout << endl;

	// 改为小顶堆实现:A B C
	priority_queue<string, vector<string>, greater<string>> p4;
	p4.push("C");
	p4.push("B");
	p4.push("A");

	cout << "p4 list 小顶堆(string): ";
	while (p4.size())
	{
		cout << p4.top() << " ";
		p4.pop();
	}
	cout << endl;
}

// 优先队列:自定义数据类型的使用 - struct
struct student
{
	int score;
	string name;
	int student_number;

	// 重载 < 运算符,用于优先级比较, 以成绩来排名
	bool operator<(const student& s) const
	{
		return this->score < s.score;	// 大顶堆
	}

	student(int s, string n, int sn) :score(s),name(n),student_number(sn){}
};

// 自定义比较函数
struct student_compare_score_greater
{
	bool operator() (const student& a, const student& b)
	{
		return a.score > b.score;	// 小顶堆
	}
};

void test_03()
{
	student s1(89, "wang", 1001001);
	student s2(76, "Li", 1001721);
	student s3(100, "Zhao", 1001321);

	priority_queue<student> p5;
	p5.push(s1);
	p5.push(s2);
	p5.push(s3);

	cout << "p5 list 大顶堆(student): " << endl << endl;
	while (p5.size())
	{
		cout << p5.top().name << " " << p5.top().score << " " << p5.top().student_number << endl;;
		p5.pop();
	}
	cout << endl;

	// 改为成绩由低到高
	priority_queue<student, vector<student>, student_compare_score_greater> p6;
	p6.push(s1);
	p6.push(s2);
	p6.push(s3);

	cout << "p6 list 小顶堆(student): " << endl;
	while (p6.size())
	{
		cout << p6.top().name << " " << p6.top().score << " " << p6.top().student_number << endl;;
		p6.pop();
	}
	cout << endl;
}

int main()
{
	// int
	test_01();
	cout << endl;

	// string
	test_02();
	cout << endl;
	
	// struct
	test_03();

	return 0;
}

运行结果如下:

p1 list 大顶堆(int): 100 10 -10
p2 list 小顶堆(int): -10 10 100

p3 list 大顶堆(string): C B A
p4 list 小顶堆(string): A B C

p5 list 大顶堆(student):

Zhao 100 1001321
wang 89 1001001
Li 76 1001721

p6 list 小顶堆(student):
Li  76  1001721
wang  89  1001001
Zhao  100  1001321

谢谢阅读

 
打赏
 本文转载自:网络 
所有权利归属于原作者,如文章来源标示错误或侵犯了您的权利请联系微信13520258486
更多>最近资讯中心
更多>最新资讯中心
更多>相关资讯中心
0相关评论

推荐图文
推荐资讯中心
点击排行
最新信息
新手指南
采购商服务
供应商服务
交易安全
关注我们
手机网站:
新浪微博:
微信关注:

13520258486

周一至周五 9:00-18:00
(其他时间联系在线客服)

24小时在线客服