C++ priority_queue

priority queue在许多的特别场合还是很实用的，优先队列是依据堆（二叉树）实现的，效率是O(lgn)，因为它带来的便利，人们可以少写很多的代码，所以学习它是有必要的。

很多时候我们会自己定义比较函数，因为标准库默认使用元素类型的<操作符来确定它们之间的优先级关系，所以如果用>在g++编译器中编译不过，在函数中operator旁边还是用<吧，或者使用()。使用”priority_queue<int,vector<int>,greater<int> >“类似的语句可以将大到小的默认顺序改成小到大（头文件是<functional>，我们也可以自己写一个比较类设定优先级。
struct cmp{
   bool operator()(type a,type b){
       return -------;
   }
};
 

#include<queue>
#include<iostream>
#include<functional>
#include<cmath>
#include<cstdio>
using namespace std;
struct cmp{
    bool operator()(int a,int b){
        return a>b;  //想让其从小到大输出,return >
    }
};
struct point{
    int x,y;
    void show(){
        printf("(%d,%d)\n",x,y);
    }
};
struct cmp2{
    bool operator()(point a,point b){
        return (abs(a.x)+abs(a.y))<(abs(b.x)+abs(b.y));
    }
};
int main(){ //优先队列默认按照从大到小的顺序排列输出
    freopen("cout.txt","w",stdout);
    int a[5]={3,5,2,1,7};
	priority_queue<int> v;
    for( int i = 0; i < 5; i++ ) {
       v.push(a[i]);
    }
    cout<<"v.size:"<<v.size()<<endl;
    while( !v.empty() ) {
       cout << v.top() << endl;
       v.pop();
    }
    cout<<"v.size:"<<v.size()<<endl;
    priority_queue<int,vector<int>,greater<int> > v2;
    //priority_queue<int,vector<int>,less<int> > v; //less 降序 ,greater 升序 
    
    for( int i = 0; i < 5; i++ ) {
       v2.push(a[i]);
    }
    cout<<"v2 从小到大输出：\n";
    while( !v2.empty() ) {
       cout << v2.top() << endl;
       v2.pop();
    }
    priority_queue<int,vector<int>,cmp> v3;
    for( int i = 0; i < 5; i++ ) {
       v3.push(a[i]);
    }
    cout<<"自定义比较的V3输出:\n";
     while( !v3.empty() ) {
       cout << v3.top() << endl;
       v3.pop();
    }
    cout<<"点输出优先级：与原点的曼哈顿距离。\n";
    point p[5]={{0,9},{12,0},{3,4},{6,5},{3,7}};
    priority_queue<point,vector<point>,cmp2> que;
    //priority_queue<point> que; 复杂的类型需要写出专门的比较类(上面的cmp2)。
    for(int i=0;i<5;i++)que.push(p[i]);
     while( !que.empty() ) {
       point tmp=que.top();
       tmp.show();
       que.pop();
    }
    return 0;
}

相关的结果：

v.size:5
7
5
3
2
1
v.size:0
v2 从小到大输出：
1
2
3
5
7
自定义比较的V3输出:
1
2
3
5
7
点输出优先级：与原点的曼哈顿距离。
(12,0)
(6,5)
(3,7)
(0,9)
(3,4)

一个应用例子：

nefu 355 合成陨石

http://acm.nefu.edu.cn/JudgeOnline/problemshow.php?problem_id=355

大意：一堆石子两两合并，合并的新石子加入原石堆中，每次合并产生的新石子质量是合并石子的质量和，且记录下来，请问最后的最小的质量和

#include <iostream>
#include <cstdio>
#include <queue>
#include <vector>
using namespace std;

int main()
{
    int n,a;
    while(cin>>n){
        priority_queue<int,vector<int>,greater<int> > que;
        for(int i=0;i<n;i++) {
            scanf("%d",&a);
            que.push(a);
        }
        if(n==1) {
            printf("%d\n",a);
            continue;
        }
        int ans=0,get=0;
        while(!que.empty()){
            get=que.top();
            que.pop();
            if(que.empty()) break;
            get=get+que.top();
            que.pop();
            ans=ans+get;
            que.push(get);
        }
        printf("%d\n",ans);
    }
    return 0;
}