14届蓝桥杯省赛 C/C++ B组 T8 整数删除（双向链表，堆）

瞬间定位一个数的左边或者右边，需要用到双向链表。
在过程中不断维护最小值，需要用到堆。

所以定义一个pair类型优先队列，每次取出堆顶进行删除，并且同时让删除元素的左右元素加上其值。

同时需要注意，在删除元素之后，我们又对两个元素进行了值的增加，所以这时候堆里对应的值和元素真实对应的值是不同的，则这时候取出来的最小值对应的下标不一定就是最小值了，如果我们扫到了真实值和堆中值不同的元素，就要重新把改变后的元素压入堆，这样才能保证完全正确。

代码：

#include<iostream>
#include<queue>
using namespace std;
const int N = 5e5 + 10;
#define pii pair<long long,int>


int n, k;
long long v[N], l[N], r[N];

int main() {
	cin >> n >> k;
	r[0] = 1, l[n + 1] = n;
	priority_queue< pii, vector<pii>, greater<pii> >heap;
	//first存value值，second存下标
	
	for (int i = 1; i <= n; i++) {
		cin >> v[i];
		l[i] = i - 1, r[i] = i + 1;
		heap.push({v[i],i});
	}
	
	while (k--) {
		FLAG:
        auto t = heap.top(); heap.pop();
        
		if (t.first != v[t.second]) {
			heap.push({ v[t.second],t.second });//如果出现了元素对不上
			goto FLAG;							//就返回去重新取
		}
		else {
			l[r[t.second]] = l[t.second];
			r[l[t.second]] = r[t.second];
			v[l[t.second]] += v[t.second];
			v[r[t.second]] += v[t.second];
		}
	}
	for (int i = r[0]; i != n+1; i = r[i]) {
		cout << v[i] << " ";
	}
	return 0;
}