优先队列
- 在头文件 < q u e u e > <queue> <queue>中
- 定义方法:priority_queue<储存的类型,vector<储存的类型>,顶堆的类型> 容器名
- less<储存的数据类型> 即使用大顶堆,即队首为最大元素
- greater<储存的数据类型> 即是用小顶堆,即队首为最小元素
例子:
priority_queue<int,vector<int>,less<int>> q;//储存int型数据
priority_queue<double,vector<double>,less<double>> q;//储存double型数据
priority_queue<string,vector<string>,less<string>> q;//储存string型数据
priority_queue<结构体名,vector<结构体名>,less<结构体名>> q;//储存结构体或者类
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<结构体名>,greater<结构体名>> q;//储存结构体或者类
成员函数:
empty() 如果优先队列为空,则返回真
pop() 删除第一个元素
push() 加入一个元素
size() 返回优先队列中拥有的元素的个数
top() 返回优先队列中有最高优先级的元素
注意:当优先队列储存的类型为pair时,先按照first的元素来比大小,first相同再按照second来比
优先队列自定义结构体排序方法:
struct node
{
int x,y;
int step;
};
priority_queue<node>q; //优先队列中元素的比较规则默认是按元素的值从大到小排序;
bool operator<(const node &a,const node &b) //括号里面是const 而且还必须是引用
{
return a.step>b.step; //从小到大排序。重载小于号。因为默认是从大到小
}
H .整数删除【蓝桥杯】
#include<iostream>
#include<queue>
using namespace std;
typedef long long LL;
//优先队列
priority_queue<pair<LL,int>,vector<pair<LL,int>>,greater<pair<LL,int>>> q;
int pre[500005],ne[500005];
LL cnt[500005],a[500005];
int main()
{
int n,k;
cin>>n>>k;
for(int i=1;i<=n;i++)
{
LL a;
cin>>a;
//存储值及下标
q.push({a,i});
//pre数组表示左边元素的下标
//ne数组表示右边元素的下标
pre[i]=i-1;
ne[i]=i+1;
}
while(q.size()>n-k)
{
LL num=q.top().first;
int id=q.top().second;
q.pop();
//cnt[i]数组记录下标为i的元素需要进行修改的值,不为0则需要修改
if(cnt[id])
{
//将修改后的加入队列
q.push({num+cnt[id],id});
cnt[id]=0;
}
else
{
//将前面的元素和后面的元素加上被删除的值
cnt[pre[id]]+=num;
cnt[ne[id]]+=num;
//修改被删除数前后两个元素的对应pre和ne值
ne[pre[id]]=ne[id];
pre[ne[id]]=pre[id];
}
}
while(!q.empty())
{
LL num=q.top().first;
int id=q.top().second;
q.pop();
a[id]=num+cnt[id];
}
for(int i=1;i<=n;i++)
{
if(a[i]) cout<<a[i]<<" ";
}
return 0;
}