BZOJ1150&&洛谷P3620 [APIO/CTSC 2007]数据备份

这应该是道贪心题

我们不难发现，一个点连电缆肯定是要和他左侧第一个或者右侧第一个连，这样是最优的，但是可能存在a,b,c,d四个点，bc是目前可连得最短边，但是连接ab和cd更优，怎么办呢？我们可以在第一次用掉bc这条边后，将ab+cd-bc作为bc边的新值加入图中，那么之后若ab+cd-bc更优的话，bc这条边就被抵消了

然后我们用一个优先队列，按相邻两两之间的距离差排序

代码

//By AcerMo
#include<cmath>
#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int M=100050; 
const int inf=1e9+7;
int n,m;
int l[M],nx[M],pr[M];
struct heap
{
	int id,len;
	bool friend operator < (heap a,heap b)
	{
		return a.len>b.len;
	}
}add,now;
priority_queue<heap>q;
inline int read()
{
	int x=0;char ch=getchar();
	while (ch>'9'||ch<'0') ch=getchar();
	while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
	return x;	
} 
signed main()
{
	n=read();m=read();
	for (int i=1;i<=n;i++) l[i]=read();
	sort(l+1,l+n+1);//坐标可能是乱的
	for (int i=1;i<n;i++)
	l[i]=l[i+1]-l[i],nx[i]=i+1,pr[i]=i-1;//做差然后设置相邻点
	nx[n-1]=0;int ans=0;
	for (int i=1;i<n;i++) 
	add.id=i,add.len=l[i],q.push(add);
	while (m--)
	{
		now=q.top();q.pop();
		if (now.len!=l[now.id]){m++;continue;}	//不同说明这个点被bc合并了
		ans+=now.len;int lp=pr[now.id],rn=nx[now.id];
		pr[nx[now.id]=nx[rn]]=now.id;//更新边界*1
        nx[pr[now.id]=pr[lp]]=now.id;//更新边界*2
		l[now.id]=(lp&&rn)?min(inf,l[lp]+l[rn]-l[now.id]):inf;//更新合并
		l[lp]=l[rn]=inf;add.id=now.id;add.len=l[now.id];q.push(add);//合并
	}
	cout<<ans;
	return 0;
}