STL的神器,优先队列

来源:互联网 发布:联合国统计数据库 编辑:程序博客网 时间:2024/05/17 07:23

在STL里有这个priority_queue,实现优先队列的结构。在优先队列中,优先级高的元素先出队列。
现在在这里说说用法吧

先看看语法:

Syntax:

In their implementation in the C++ Standard Template Library,priority queues take three template parameters:1
2 template < class T, class Container =vector,
          class Compare = less > class priority_queue;


Where the template parameters have the following meanings:
T: Type of the elements.
Container: Type of the underlying container object used to storeand access the elements.
Compare: Comparison class: A class such that the expressioncomp(a,b), where comp is an object of this class and a and b areelements of the container, returns true if a is to be placedearlier than b in a strict weak ordering operation. This can eitherbe a class implementing a function call operator or a pointer to afunction. This defaults to less, which returns the same as applyingthe less-than operator (a
The priority_queue object uses this expression when an element isinserted or removed from it (using push or pop, respectively) togrant that the element popped is always the greater in the priorityqueue.

可以自定义一个比较类,Compare

其实就三种用法

第一种,直接使用默认的。

它的模板声明带有三个参数,priority_queue
Type 为数据类型, Container 为保存数据的容器,Functional 为元素比较方式。
Container 必须是用数组实现的容器,比如 vector, deque 但不能用 list.
STL里面默认用的是 vector. 比较方式默认用 operator< ,所以如果你把后面俩个
参数缺省的话,优先队列就是大顶堆,队头元素最大。
看例子


priority_queue qi;

    inta[len] = {3,5,9,6,2};

   priority_queue qi;
    for(i = 0; i< len; i++)
       qi.push(a[i]);

    for(i =0; i < len; i++)
    {
       cout<<qi.top()<<"";
       qi.pop();
    }

通过<操作符可知在整数中元素大的优先级高。
故例子中输出结果为:9 6 5 3 2

第二种:

第二种方法:
在示例1中,如果我们要把元素从小到大输出怎么办呢?
这时我们可以传入一个比较函数,使用functional.h函数对象作为比较函数。

如果要用到小顶堆,则一般要把模板的三个参数都带进去。
STL里面定义了一个仿函数greater<>,对于基本类型可以用这个仿函数声明小顶堆
priority_queue, greater >qi2;

对于自定义类型,则必须自己重载 operator< 或者自己写仿函数

#include
#include

using namespace std;

struct Node{
    int x,y;
    Node( int a=0, int b= 0 ):
       x(a), y(b) {}
};

bool operator<( Node a, Node b ){
    if( a.x==b.x ) return a.y> b.y;
    returna.x> b.x;
}

int main(){
   priority_queue q;
   
    for( int i=0; i< 10; ++i )
    q.push(Node( rand(), rand() ) );
   
    while(!q.empty() ){
       cout << q.top().x<< ' '<< q.top().y<< endl;
       q.pop();
    }
   
   getchar();
    return0;
}

或者这样定义也是能达到效果的:

struct Node{
    int x,y;
    Node( int a=0, int b= 0 ):
       x(a), y(b) {}

    friendoperator<( Node a, Node b ){
    if( a.x==b.x ) return a.y> b.y;
    returna.x> b.x;

    }
};

自定义类型重载 operator< 后,声明对象时就可以只带一个模板参数。
但此时不能像基本类型这样声明
priority_queue, greater >;
原因是 greater 没有定义,如果想用这种方法定义
则可以按如下方式

例子:

#include
#include

using namespace std;

struct Node{
    int x,y;
    Node( int a=0, int b= 0 ):
       x(a), y(b) {}
};

struct cmp{
    booloperator() ( Node a, Node b ){
       if( a.x== b.x ) return a.y> b.y;
       
       return a.x> b.x; }
};

int main(){
   priority_queue, cmp> q;
   
    for( int i=0; i< 10; ++i )
    q.push(Node( rand(), rand() ) );
   
    while(!q.empty() ){
       cout << q.top().x<< ' '<< q.top().y<< endl;
       q.pop();
    }
   
   getchar();
    return0;
}

还有一点要注意的是priority_queue中的三个参数,后两个可以省去,因为有默认参数,不过如果,有第三个参数的话,必定要写第二个参数。

转自:http://hi.baidu.com/einstein17/item/dad993a7bff828258819d3c8


一个简单的测试:#include<iostream>

#include<queue>

using namespace std;

const int len = 5;

int main()

{

priority_queue<int>qi;

int a[len] = {3,5,9,6,2};

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

qi.push(a[i]);

while(!q.empty())

{

cout<<qi.top()<<"";

qi.pop();

}

cout<<endl;

}