转自:http://blog.csdn.net/senyelicone/article/details/51850284
1 #include<cstdio> 2 #include<queue> 3 using namespace std; 4 5 priority_queue<int> da; //大根堆 6 priority_queue<int,vector<int>,greater<int> > xiao; //小根堆,最后的两个“>”之间要有空格,vector不用单另开头文件。至于greater是什么意思,我并不知道……
C++中优先队列(priority queue<>)使用头文件queue,需要声明using namespace std;
常用来代替堆,每次直接加入新数即可,自动维护大小顺序,使用很方便。
以大根堆为例,q.top( )是队中最大的数。常用操作还有:q.push_back(x),q.pop( ) …
上面的程序是基本的建立队列,可以建立大根堆或小根堆,不必强制记忆,每次调试一下就可以了。
(以上是优先队列中常用内容,下面是转载的很神奇的用法。(感觉用不到))
#include<iostream> #include<functional> #include<queue> using namespace std; struct node { friend bool operator< (node n1, node n2) { return n1.priority < n2.priority; //"<"为从大到小排列,">"为从小到大排列 } int priority; int value; }; int main() { const int len = 5; //也可以写在函数内 int i; int a[len] = {3,5,9,6,2}; //示例1 priority_queue<int> qi; //普通的优先级队列,按从大到小排序 for(i = 0; i < len; i++) qi.push(a[i]); for(i = 0; i < len; i++) { cout<<qi.top()<<" "; qi.pop(); } cout<<endl; //示例2 priority_queue<int, vector<int>, greater<int> > qi2; //从小到大的优先级队列,可将greater改为less,即为从大到小 for(i = 0; i < len; i++) qi2.push(a[i]); for(i = 0; i < len; i++) { cout<<qi2.top()<<" "; qi2.pop(); } cout<<endl; //示例3 priority_queue<node> qn; //必须要重载运算符 node b[len]; b[0].priority = 6; b[0].value = 1; b[1].priority = 9; b[1].value = 5; b[2].priority = 2; b[2].value = 3; b[3].priority = 8; b[3].value = 2; b[4].priority = 1; b[4].value = 4; for(i = 0; i < len; i++) qn.push(b[i]); cout<<"优先级"<<' '<<"值"<<endl; for(i = 0; i < len; i++) { cout<<qn.top().priority<<' '<<qn.top().value<<endl; qn.pop(); } return 0; }
对于队列里元素为一个结构体类型,按照某一个属性排序,就需要对比较函数进行重载
小结一下:
1、若是定义值类型对象,如:
int main() { priority_queue<node> qn; node n1; n1.a = 9; node n2; n2.a = 2; node n3; n3.a = 50; qn.push(n1); qn.push(n2); qn.push(n3); int size = qn.size(); for ( int i = 0; i < size; i++) { cout << qn.top().a << endl; qn.pop(); } return 0; }
则直接运算符重载即可:
class node { public : int a; node(){} node( int x):a(x){} friend bool operator<( const node ne1, const node ne2)//参数也可以为引用,值传递 { if (ne1.a > ne2.a) { return true ; } else { return false ; } } };
其中在c++primer第三版 中文版中关于操作符重载有如下描述:
"程序员只能为类类型或枚举类型的操作数定义重载操作符我们可以这样来实现把重
载操作符声明为类的成员或者声明为名字空间成员同时至少有一个类或枚举类型的参数
按值传递或按引用传递"
因此不可用指针类型的参数;
2、如果想定义一个指针类型的优先队列,那就不可这么简单的定义了,你需要自定义一个自己的比较函数,在priority_queue的模板函数中,我们可以利用这样一个template<class _Ty, class _Container = vector<_Ty>, class _Pr = less<typename _Container::value_type> >模板,在我们的程序代码中,则需要自己定义一个类来定义第三个模板参数,如:
class nodePointerComparer
{
public :
nodePointerComparer(){}
bool operator ()( const node* ne1, const node* ne2) const
{
return ne1->lessthan(ne2);
}
};
当然在这之前我们的类Node要重新书写
class node
{
public :
int a;
node(){}
node( int x):a(x){}
bool lessthan( const node* pnode) const
{
return a < pnode->a;
}
};
这样在main函数中就可以定义指针类型的优先队列了:
int main()
{
priority_queue <node*, vector <node*>, nodePointerComparer> qn;
node *n1 = new node(90);
node *n2 = new node(2);
node *n3 = new node(50);
qn.push(n1);
qn.push(n2); /span>
>
qn.push(n3);
int size = qn.size();
for ( int i = 0; i < size; i++)
{
cout << qn.top()->a << endl;
qn.pop();
}
return 0;
}
疑问之处:如果你使用第一种值传递的操作符重载,来实现第二种的指针类型优先队列,是不会达到想要的结果的,个人理解是因为在指针类型的优先队列中找“<”运算符的时候,重载的不是我们写的值传递friend bool operator<(const node ne1,const node ne2)//
也就是没有找到指针类型的"<"重载,所有不会达到优先队列的效果。