H.整数删除【蓝桥杯】优先队列+双向链表

优先队列

• 在头文件$<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;
}