题目描述
给定一个长度为 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
题意描述
给定一个长度为N 的整数数列:
A
1
,
A
2
…
A
N
A_1,A_2…A_N
A1,A2…AN。
你要重复以下操作K 次:每次选择数列中最小的整数(如果最小值不止一个,选择最靠前的),将其删除。并把与它相邻的整数加上被删除的数值。
输出K 次操作后的序列。
双向链表 | 最小堆
由于要进行大量的删除操作, 不难想到可以使用链表.
而本题需要动态的求最小值, 显然可以使用堆.
每次从堆中取出最小值的下标, 然后在链表中删除它.
但本题特殊点在于将其删除。并把与它相邻的整数加上被删除的数值, 所以会导致还在堆中的元素的权的变化.
我们可以注意到, 每次删除操作只会让一些元素变大, 而不会让元素变小. 也就是, 可能会让原本的最小值变成不是最小值.
因此我们取出堆中的最小值时, 需要将此元素的排序权和实际的值进行对比, 如果实际的值变大了, 则当前元素并不一定是最小值, 需要重新放回堆中.
参考代码
每次删除操作最多会让两个元素的值变化, 因此从堆中取出的次数是k的线性, 时间复杂度为O(n + k) \log n
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
const int N = 5e5 + 10;
ll v[N], l[N], r[N];
//双向链表的删除操作, 删除x结点
void del(int x) {
r[l[x]] = r[x], l[r[x]] = l[x];
v[l[x]] += v[x], v[r[x]] += v[x];
}
int main () {
int n, k; cin >> n >> k;
//最小堆, 堆中的元素是{权值, 结点下标}
priority_queue<pair<ll, int>, vector<pair<ll, int>>, greater<pair<ll, int>>> h;
//输入并构造双向链表
r[0] = 1, l[n + 1] = n;
for (int i = 1; i <= n; i ++)
cin >> v[i], l[i] = i - 1, r[i] = i + 1, h.push({v[i], i});
while (k --) {
auto p = h.top(); h.pop();
//如果v发生变化, 则目前的元素不一定是最小值, 需要重新放入堆中
if (p.first != v[p.second]) h.push({v[p.second], p.second}), k ++;
else del(p.second);
}
//输出链表剩余的元素
int head = r[0];
while (head != n + 1) {
cout << v[head]<< " ";
head = r[head];
}
return 0;
}