4961. 整数删除 - AcWing题库
题目描述
分析
注：如果要进行大量的删除操作可以使用链表
动态求最小值使用堆，每次从堆中取出最小值的下标然后在链表中删除
注意long long
代码解释：
while(k --){auto t = q.top();q.pop();res = t.first;i = t.second;if(res != v[i])q.push({v[i], i});else del(i); }eg. 2 3 4此时这三个数的下标分别为1 2 3
第一步：在q的队列中加入2， 3， 4，第一次k --进行del操作，使v[2] == 5
第二部：q.top() == 3发现3对应下标为2， v[2]原本为3，但是上一步使其变为了5，故此时需要重新将5加入队列，当然，此时k不算进行了一次操作，需要k ++(因为这一步只是将上一步两边加数的操作进行了完善）
#include<bits/stdc++.h>
using namespace std;
const int N = 5e5 + 10;
typedef long long ll;
typedef pair<ll, ll> PII;
priority_queue<PII, vector<PII>, greater<PII>>q;
ll n, k, v[N], l[N], r[N];
void del(ll x)
{r[l[x]] = r[x], l[r[x]] = l[x];v[l[x]] += v[x], v[r[x]] += v[x];
}
int main()
{ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);cin >> n >> k;r[0] = 1, l[n + 1] = n;//初始化左右端点的下标，将0后的下标赋于1，将n + 1左边的下标赋予n for(int i = 1; i <= n; i ++){cin >> v[i];//下标i对应的值为v[i]l[i] = i - 1;//建立双链表 r[i] = i + 1; q.push({v[i], i});//将值和对应下标存入优先队列 }while(k --){auto t = q.top();q.pop();ll res = t.first;ll i = t.second;if(res != v[i]){q.push({v[i], i});k ++;}else del(i); }for(int i = r[0]; i != n + 1; i = r[i]){cout << v[i] << ' ';}cout << '\n';return 0;
}
