C++ 优先队列的基本使用方法

[cpp] view
plain copy

 





#include<cstdio>  

#include<queue>  

using namespace std;  

  

priority_queue<int> da;  //大根堆  

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( ) …

上面的程序是基本的建立队列，可以建立大根堆或小根堆，不必强制记忆，每次调试一下就可以了。

（以上是优先队列中常用内容，下面是转载的很神奇的用法。（感觉用不到））

[cpp] view
plain copy

 





#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<<"优先级"<<'\t'<<"值"<<endl;  

    for(i = 0; i < len; i++)  

    {  

        cout<<qn.top().priority<<'\t'<<qn.top().value<<endl;  

        qn.pop();  

    }  

    return 0;  

}  

对于队列里元素为一个结构体类型，按照某一个属性排序，就需要对比较函数进行重载

小结一下：

1、若是定义值类型对象，如： 

[cpp] view
plain copy

 





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;  

}   

则定义优先级的时候，直接在类中写个friend 的操作符重载函数即可： 

[cpp] view
plain copy

 





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)//

也就是没有找到指针类型的"<"重载，所有不会达到优先队列的效果。

转载自：点击打开链接