基础用法
头文件:#include<queue>
默认创建大根堆:
priority_queue<int> q;//储存int型数据
priority_queue<double> q;//储存double型数据
priority_queue<string> q;//储存string型数据
priority_queue<结构体名> q;//储存结构体或者类
常用方法:
empty() 如果优先队列为空,则返回真
pop() 删除第一个元素
push() 加入一个元素
size() 返回优先队列中拥有的元素的个数
top() 返回优先队列中有最高优先级的元素
创建小根堆:
priority_queue<int,vector<int>,greater<int>> q;//储存int型数据
priority_queue<double,vector<double>,greater<double>> q;//储存double型数据
priority_queue<string,vector<string>,greater<string>> q;//储存string型数据
priority_queue<自定义结构体名,vector<自定义结构体名>,Compare> q;//储存结构体或者类
//重载比较方法
struct Compare {
bool operator()(const vector<int>& a, const vector<int>& b) {
return a[1] < b[1]; // < 是大根堆 > 是小根堆
}
};
算法题示例
思路:这里创建一个优先队列保存奶酪差值:第一只老鼠得分-第二只老鼠得分 ,采用贪心思想选取最高的前k个值。
class Solution {
public:
struct Compare {
bool operator()(const vector<int>& a, const vector<int>& b) {
return a[1] < b[1]; // 按照 reward1-reward2 的差值降序排序
}
};
int miceAndCheese(vector<int>& reward1, vector<int>& reward2, int k) {
int len = reward1.size();
priority_queue<vector<int>, vector<vector<int>>, Compare> q;
int i = 0;
for(;i<len;i++){
q.push({i, reward1[i] - reward2[i]}); // 向优先队列中添加元素
}
int res = 0;
for(i = 0;i<k;i++){
vector<int> t = q.top();
res += reward1[t[0]];
q.pop();
reward2[t[0]] = 0;
}
for(i=0;i<len;i++){
res += reward2[i];
}
return res;
}
};