priority_queue用法

本文介绍如何在C++中使用STL的priority_queue,并自定义比较方式。通过一个具体例子展示了如何创建一个自定义的数据结构Node以及比较器cmp,并应用于优先队列中。

看了一上午书换了一上午皮肤,终于会了。

模板原形:

template <class T, class Container = vector<T>,
  class Compare = less< typename Container::value_type> > class priority_queue;
T是数据类型,Container 是保留数据的容器,Functional是元素的比较方式。

Container必须是用数组实现的容器,比如vector,deque但是不能是list。

STL 里面容器默认的用的是vector,比较方式默认用 < (即越小优先级越低)

下面代码是我自己写的,自定义比较方式的用法之一:

#include <iostream>
#include <queue>
#include <vector>
using namespace std;
struct Node{
	int x;
	int y;
	friend ostream & operator << (ostream & os,const Node & tmep);
	friend istream & operator >> (istream & is,Node & temp);
};
struct cmp{
	bool operator () (const Node & a,const Node & b)const{  //据说这个叫仿函数,我也不知道是个什么东西,先这么用着吧;
		return (a.x + a.y)>(b.x + b.y);
	}
};
istream & operator >> (istream & is,Node & temp){
	is>>temp.x>>temp.y;
	return is;
}
ostream & operator << (ostream & os,const Node & temp){
	os<<temp.x<<' '<<temp.y;
	return os;
}
priority_queue<Node,vector<Node>,cmp> q;
int main(){
//	freopen("Data.txt","r",stdin);
	Node temp;
	while(cin>>temp)
		q.push(temp);
	while(!q.empty()){
		cout<<q.top()<<endl;
		q.pop();
	}
	return 0;
}

C++中的`priority_queue`是一种优先队列,其默认实现是最大堆。它允许以对数时间复杂度插入元素,并且每次操作后队列顶部始终是当前所有元素中优先级最高的一个。对于基本类型,默认的排序准则是通过小于运算符(`operator <`)来定义降序排列,而对于自定义类型(如结构体),需要显式地提供比较逻辑。 ### 使用结构体进行自定义排序 当使用结构体作为`priority_queue`的元素时,需要为其定义排序规则。这可以通过以下两种方式之一完成: 1. **重载小于运算符 (`operator <`)** 在结构体内部重载`<`运算符,这样可以直接在默认模板参数下使用该结构体。例如: ```cpp struct Person { int age; bool operator<(const Person& other) const { return age < other.age; // 最大堆,年龄大的排前面 } }; std::priority_queue<Person> pq; ``` 2. **自定义比较函数对象** 如果希望使用不同的排序准则或者不便于修改结构体定义,则可以传递一个比较函数或函数对象给`priority_queue`的模板参数列表。例如: ```cpp struct Person { int age; }; struct ComparePerson { bool operator()(const Person& a, const Person& b) { return a.age < b.age; // 最大堆 } }; std::priority_queue<Person, std::vector<Person>, ComparePerson> pq; ``` ### 示例代码 下面是一个完整的示例,展示如何使用结构体与自定义比较器来创建一个基于年龄排序的优先队列: ```cpp #include <iostream> #include <queue> struct Person { int age; }; // 自定义比较器 struct ComparePerson { bool operator()(const Person& a, const Person& b) { return a.age < b.age; // 年龄大的优先 } }; int main() { std::priority_queue<Person, std::vector<Person>, ComparePerson> pq; // 插入数据 pq.push({30}); pq.push({25}); pq.push({40}); // 输出并移除队列顶部元素 while (!pq.empty()) { std::cout << pq.top().age << " "; pq.pop(); } return 0; } ``` 在这个例子中,输出将会是按年龄从高到低排序的结果:`40 30 25`。 ### 注意事项 - `priority_queue`底层通常采用`vector`作为容器,并利用堆算法维护堆属性。 - 默认情况下,`priority_queue`构造的是最大堆;若要构造最小堆,可通过指定第三个模板参数为`std::greater<>`[^4]。 - 当处理复杂类型的元素时,确保比较逻辑正确无误,以免导致错误的排序结果。 以上方法适用于大多数情况下的需求,能够灵活地适应不同场景下的自定义排序要求。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值