8. 整数删除-链表&&优先队列

8.整数删除 - 蓝桥云课 (lanqiao.cn)

#include <bits/stdc++.h>
#define int long long
#define endl '\n'
using namespace std;
//时间复杂度我觉得应该是nlogn
//因为k==n，优先队列插入和删除都是logn，执行k次操作，就是klogn==nlogn
void solve(){
	int n,k;
//	q优先队列，存的是数的值和数的编号(索引值)，小根堆，每一次都是最小值出列
	priority_queue<pair<int,int>,vector<pair<int,int>>,greater<pair<int,int>>> q;
	cin>>n>>k;
//	l数组 存 当前点 的 左侧 最近 的 没有被删除 的元素的 编号
//	r数组 存 当前点 的 右侧 最近 的 没有被删除 的元素的 编号
//	也就是链表，只不过用俩数组来存左右邻居是谁
//	st数组用来存当前点的最新值
	vector<int> l(n),r(n),st(n);
	for(int i=0;i<n;i++){//初始化
		int temp;
		cin>>temp;
		q.push({temp,i});
		st[i]=temp;
		l[i]=i-1;
		r[i]=i+1;
		
	}
	l[0]=-1;
	r[n-1]=-1;
	while(k){//执行k次操作
		int min,pos;
		min=q.top().first;//因为这道题优先队列用的小根堆，每次top都是取最小值
		pos=q.top().second;
		q.pop();//删除元素，删除优先级最高的，也就是最小值
		if(min!=st[pos]){//如果当前最小值和当前点最新值不相等
		//说明得更新，然后直接快进到下一次循环，记住刚才那几步操作K没有减一
			q.push({st[pos],pos});
			continue;
		}
		k--;//到这里才自减
		//if判断是为了防止动态数组下标是-1的情况，-1的话数组越界了
		//下面这几行是把当前最小值和左右两点的箭头指向去掉，让当前点左右两点互相指向(双向链表的思想)
		//还得把 当前点 左右两点 的 数值加上当前点的数值
		if(l[pos]!=-1){
			r[l[pos]]=r[pos];
			st[l[pos]]+=min;
		}
		if(r[pos]!=-1){
			l[r[pos]]=l[pos];
			st[r[pos]]+=min;
		}
//		把当前点赋值-1，此步是为了最后输出的时候，好判断输出
		st[pos]=-1;
	}
	for(int i=0;i<n;i++){
		//如果当前点不是-1就输出
		if(st[i]!=-1) cout<<st[i]<<" ";
	}
}
signed main(){
	ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);
	solve();
	return 0;
}