priority_queue在结构体中的使用

本文旨在介绍priority_queue对于结构体的使用，利用友元函数进行重载

---

一、priority_queue简介

priority_queue是基于堆等数据结构配以一定算法存储于STL容器中供以使用的容器。需要配以头文件。

二、使用

1.对于普通数据而言，可以如下列写法

代码如下（示例）：

#include<iostream>
#include<queue>
using namespace std;
priority_queue<int>q;
int main()
{
	//注意优先队列无push_back与front
	q.push(3);
	q.push(4);
	q.push(5);
	while(!q.empty())
	{
		cout<<q.top()<<endl;
		q.pop();
	}
	cout<<"end";
	return 0;
}

输出结果应该为：
5
4
3
end
由大到小输出
若想有小到大输出，可以如下设置

priority_queue<int,vector<int>,greater<int> >q;
//若按第一种写法则会默认为 priority_queue<int,vector<int>,less<int> >q;

2.结构体

代码如下（示例）：

#include<iostream>
#include<queue>
using namespace std;
struct node{
	int x;
	int y;
	node(int a,int b){
		x=a;
		y=b;
	}
	friend bool operator <(const node xx,const node yy)
	//重载小于号  注意不能重载大于号
	{
	//该处设置比较方法，示例为优先比较x，x小的优先，再比较y，y小的优先
		if(xx.x!=yy.x)
			return xx.x>yy.x;
		else{
			return xx.y>yy.y;	
			}	
	} 
};
priority_queue<node>q;
int main()
{
	q.push(node(3,4));
	q.push(node(4,5));
	q.push(node(4,7));
	q.push(node(3,3));
	while(!q.empty())
	{
		cout<<q.top().x<<"->"<<q.top().y<<endl;
		q.pop();
	}
	cout<<"end";
	return 0;
}

输出如下：
3->3
3->4
4->5
4->7
end

---

总结

重载小于号与在优先队列中进行比较的方法有很多，这里只展示一种，对于算法使用而言，必须掌握至少一种，上述方法为笔者认为值得掌握的方法。