蓝桥杯2023年第十四届省赛真题-整数删除

题目描述

给定一个长度为 N 的整数数列：A1, A2, . . . , AN。你要重复以下操作 K 次：

每次选择数列中最小的整数（如果最小值不止一个，选择最靠前的），将其删除。并把与它相邻的整数加上被删除的数值。输出 K 次操作后的序列。

输入格式

第一行包含两个整数 N 和 K。

第二行包含 N 个整数，A1, A2, A3, . . . , AN。

输出格式

输出 N − K 个整数，中间用一个空格隔开，代表 K 次操作后的序列。

样例输入

复制

5 3
1 4 2 8 7

样例输出

复制

17 7

提示

数列变化如下，中括号里的数是当次操作中被选择的数：

[1] 4 2 8 7

5 [2] 8 7

[7] 10 7

17 7

对于 20% 的数据，1 ≤ K < N ≤ 10000。

对于 100% 的数据，1 ≤ K < N ≤ 5 × 105，0 ≤ Ai ≤ 108。

解题思路：也没啥就是用个链表和优先队列废话不多说直接上代码

#include<bits/stdc++.h>
using namespace std;
#define val first
#define pos second
typedef long long ll;
typedef pair<ll,int> p;
const int N=5e5+5;
int n,k,pre[N],nex[N];
ll a[N];
priority_queue<p>q;
int main()
{
    cin>>n>>k;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        pre[i]=i-1;
        nex[i]=i+1;
        q.push({-a[i],-i});
    }
    pre[1]=-1;
    nex[n]=-1;
    while(k--)
    {
        p now;
        do{
            now=q.top(); q.pop();
            now.val=-now.val;now.pos=-now.pos;
        }while(a[now.pos]!=now.val);
        int PRE=pre[now.pos];
        int NEX=nex[now.pos];
        if(PRE!=-1){
            a[PRE]+=now.val;
            q.push({-a[PRE],-PRE});
            nex[PRE]=NEX;
        }
        if(NEX!=-1){
            a[NEX]+=now.val;
            q.push({-a[NEX],-NEX});
            pre[NEX]=PRE;
        }
        a[now.pos]=-1;
      
    }
      
        for(int i=1;i<=n;i++)
        {
            if(a[i]!=-1)
            cout<<a[i]<<" ";
        }
        
}