#优先队列
优先队列就是STL的#include<queue>
里面的 priority_queue
,它的原理就是通过堆,这里来一张书上的图
也就是每push() 一个元素,这个priority_queue就会自动排序,也就是通过堆排序中的调整堆,复杂度也是相当好的O(logn)
使用的时候要注意,如果是基础类型,它默认是降序所以要想升序有点小改动,而且对于基础类型,库函数中提供了可以直接调用的函数
升序 : priority_queue<int, vector<int>, greater<int> > list;
降序 : priority_queue<int, vector<int>, less<int> > list;
其中要注意的是最后那两个 > 不能贴在一起,因为就和 >> 冲突了
但是如果遇到自己想排序的结构体或者class的时候,就只有自己动手丰衣足食了。。。
#include<iostream>
#include<queue>
using namespace std;
typedef struct Node{
int x, y;
Node(int xx, int yy){x = xx; y = yy;}
bool operator < (Node const &a) const {
return x > a.x;
}
}Node;
int main(){
priority_queue<Node> list;
list.push(Node(5,3));
list.push(Node(1,1));
list.push(Node(3,3));
while(!list.empty()){
Node temp = list.top();
list.pop();
cout << "(" << temp.x << " " << temp.y << ") ";
}
return 0;
}
这里有几点要注意的地方:
1 、重载运算符的时候只能重载 <
符号
2 、申明方法参数只有一个,而且是const 引用类型
3 、方法参数后面要加一个 const
,这是表明此方法不能修改类中的成员变量
第一题poj3253
这道题遇到过,就是哈夫曼树的思想,不停找当前数组中最小的两个,然后将它们拿出来然后相加,再讲相加的和又放入到数组中,我之前就用学校头学的哈夫曼树的方法切套这道题还超时了。。。
这里就用priority_queue
来解决这道题
#include<iostream>
#include<queue>
using namespace std;
int length;
int main(){
priority_queue<long long, vector<long long>, greater<long long> > list;
long long input;
cin >> length;
for(int i = 0; i < length; i++){
cin >> input;
list.push(input);
}
long long a, b, answer = 0;
while(list.size() > 1){
a = list.top(); list.pop();
b = list.top(); list.pop();
answer += a + b;
list.push(a + b);
}
cout << answer << endl;
return 0;
}
第二题poj2431
这道题还是出了很多问题,我开始想的是不停的找从当前位置向目的地走,其中找出最大的那个可以加油的加油站,然后下一次就直接定位到这个加油站位置,又继续那个过程,但是紧到WA,我后头突然想到一个测试数据
10
1 0
2 1
3 1
4 1
5 1
6 1
7 1
8 1
9 5
10 4
11 3
对于这组数据来说,如果还是按我刚才那个想法就有问题了,因为他每次都切找最大的,那他第一次肯定是找9这个位置的加油站,然后下一次的时候就从9这个位置开始继续往前找,然后前面都是1了,但其实应该在 10 这个位置也停下加油,然后就可以直接到目的地了,所以我就把找过的所有都放到一个优先队列里面,然后如果可以到更前面的加油站时,我就又把新的加油站放进这个优先队列中,然后不停找优先队列中最大的那个
#include<iostream>
#include<queue>
#include<algorithm>
using namespace std;
typedef struct Node{
int index;
int dist, fuel;
Node(){}
bool operator < (Node const &a) const {
return fuel < a.fuel;
}
}Node;
Node shuzu[10005];
int dist, fuel;
int jishu;
bool cmp(Node a, Node b){
return a.dist < b.dist;
}
int main(){
cin >> jishu;
for(int i = 0; i < jishu; i++)
cin >> shuzu[i].dist >> shuzu[i].fuel;
sort(shuzu, shuzu+jishu, cmp);
for(int i = 0; i < jishu; i++) shuzu[i].index = i;
cin >> dist >> fuel;
int i = jishu - 1;
int answer = 0;
int flag = 1;
priority_queue<Node> list;
while(dist-fuel > 0){
for(; i>=0 && shuzu[i].dist>=dist-fuel; i--){
if(shuzu[i].index == -1) continue;
list.push(shuzu[i]);
}
if(list.empty()) {flag = 0; break;}
Node temp = list.top();
list.pop();
if(dist >= temp.dist){
fuel = fuel - dist + temp.dist + temp.fuel;
dist = temp.dist;
}else{
fuel = fuel + temp.fuel;
}
shuzu[temp.index].index = -1;
answer ++;
}
if(flag == 1){
cout << answer << endl;
}else{
cout << "-1" << endl;
}
return 0;
}