用priority_queue创建最小堆

标准库里的priority_queue默认创建的是最大堆，要创建最小堆的话，需要下面的形式。

priority_queue<int,vector<int>,greater<int>> minheap;//最小堆
priority_queue<int,vector<int>,less<int>> maxheap;//最大堆

默认的就是最大堆，可以直接像下面那样写

priority_queue<int> maxheap;//最大堆

例子：

#include <iostream>
#include <queue>
using namespace std;
int main()
{
    priority_queue<int,vector<int>,greater<int>> pqueue;
    int a[] = { 2,14,53,1,45 };
    for (int i = 0; i < 5; i++)
    {
        pqueue.push(a[i]);
    }
    for (int i = 0; i < 5; i++)
    {
        cout << pqueue.top() << endl;
        pqueue.pop();
    }
    return 0;
}

 

 

对于自己构建的类型，还需要对运算符进行重载。需要最小堆的时候，需要对大于号重载；需要最大堆的时候，需要对小于号重载。

比如需要最小堆

struct Node {
     double value;
     int idx;
     Node (double v, int i): value(v), idx(i) {}
     friend bool operator > (const struct Node &n1, const struct Node &n2) {             
     return n1.value > n2.value;
     } 
};

//新建一个最小堆 
 priority_queue<Node, vector<Node>, greater<Node>> minheap;