优先队列priority_queue使用方法

优先队列priority_queue

• 头文件 :

#include <queue>

• 函数成员：

empty();//判断是否为空  
push(Elem e);//队列尾部增加一元素  
pop();//队列头部数据出队  
top();//返回头部数据  
size();//返回栈中元素个数

• 说明：优先队列是一个容器适配器，它使用其它容器作为底层容器，并提供一系列访问元素的函数。优先队列中的元素从底层容器的“尾部（back）”弹出，即是从队列的顶部（top）弹出。底层容器需要满足的条件是可以通过迭代器随机访问其中的元素并且支持以下的操作：

  • front()
  • push_back()
  • pop_back()

故此，容器vector和deque可以作为底层容器，在创建优先队列时若没有声明使用何种容器，则默认使用vector。

• 声明：priority_queue<Type, Container, Functional>
  Type — 数据类型
  Container — 保存数据的容器，必须是用数组实现的容器，比如 vector, deque 但不能用 list. STL里面默认用的是 vector
  Functional —元素比较方式，默认用 operator<，如果前两个参数优先队列就是大顶堆，队头元素最大。

使用：

方法1：

priority_queue<int> qi; // 队头最大，容器vector

方法2：

priority_queue<int, vector<int>, greater<int>> qi2;

greater头文件：

#include <functional>    // std::greater

• greater为STL定义的仿函数，容器内降序排列 (x > y)，但此队头即为vector尾部，所以在此为降序排列
• less在容器内为升序排列 (x < y)，但此队头即为vector尾部，所以在此为升序排列
• 如果是pair类型，则以pair.first为key进行排序

1. 如果是自定义类型，可在类中友元重载运算符：

struct Node{
    int x, y;
    Node( int a= 0, int b= 0 ):
        x(a), y(b) {}
        
    friend operator<( Node a, Node b ){
    if( a.x== b.x ) return a.y> b.y; // 如果x相同则比较y
    return a.x> b.x;
    }
};

2. 或在类外重载运算符：

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;
    return a.x> b.x;
}

priority_queue<Node> q;

3. 也可以自定义比较函数:

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

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

priority_queue<Node, vector<Node>, cmp> q;