P3620-[APIO/CTSC2007]数据备份【贪心,堆,链表】

正题

题目链接:https://www.luogu.com.cn/problem/P3620

---

题目大意

一条线上有$n$个位置，选出$k$对使得它们的距离差之和最小。

---

解题思路

因为一定是连接相邻的最优，那么可以在差分数组上做，相当于我们在一个差分数组上选择一些不相邻的数使得它们和最小。

考虑贪心，我们对于一个最小的位置$x$选择了之后，更优的情况可能是选择$x$相邻的两个然后不选择$x$。那么就可以删除这三个数然后加入一个$d_{x-1}+d_{x+1}-d_x$。

用链表+堆维护即可。时间复杂度$O(n\log n)$

---

$code$

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
#define mp(x,y) make_pair(x,y)
using namespace std;
const ll N=1e5+10;
struct del_d{
	priority_queue<pair<ll,ll> >q1,q2;
	void push(pair<ll,ll> x)
	{q1.push(x);return;}
	void pop(pair<ll,ll> x)
	{q2.push(x);return;}
	pair<ll,ll> top(){
		while(!q2.empty()&&q1.top()==q2.top())
			q1.pop(),q2.pop();
		return q1.top();
	}
}q;
ll n,k,d[N],last[N],next[N],ans;
void del(ll x){
	if(!x)return;
	q.pop(mp(-d[x],x));
	last[next[x]]=last[x];
	next[last[x]]=next[x];
	return;
}
int main()
{
	scanf("%lld%lld",&n,&k);
	ll las;
	for(ll i=0;i<n;i++){
		ll x;scanf("%lld",&x);
		if(i!=0)d[i]=x-las,q.push(mp(-d[i],i));
		las=x;
	}
	n--;
	for(ll i=1;i<=n;i++)
		last[i]=i-1,next[i]=i+1;
	next[n]=0;d[0]=1e18;
	while(k--){
		ll x=q.top().second;ans+=d[x];
		q.pop(q.top());d[x]=d[last[x]]+d[next[x]]-d[x];
		del(last[x]);del(next[x]);q.push(mp(-d[x],x));
	}
	printf("%lld\n",ans);
	return 0;
}