BZOJ1150 数据备份

首先一定取的每一段都连接着相邻的楼

我们可以每次取最小的一段用堆维护

但是这样发现过不了样例，因为一栋楼只能被连接一次

考虑样例

转化为2 1 2 6中取俩个但不能相邻

在取1的时候可以加入2+2-1=3

以后取3表示不取1取两边

重点是边境

边界同样要考虑

比如1 2 3 4

取完1以后加入1显然是有问题的（少加了个数）

我们其实可以直接舍弃2，可以证明无论在什么情况下，取2一定没有取1优

取2后能取的其它数取1后也能取

故把边境key【0】key【n】赋值MAX 这样就自动舍弃2了

再如2 1 3 5 7 8

取 1 加 4

取 4 加 1

取 1 加 6

问题来了1+4+1=1+5 俩个数计算了三次

而key【0】=MAX时

取 1 加 4

取 4 加 MAX

取 7 加 MAX

1+4+7=2+3+7

正确

key【0】=MAX相当于舍弃了5 而 取5后能取的其它数取3后也能取 （1在过程中被删表示不是最优 不算1 ）

故代码注意边境MAX

若环就不用

若链取MAX就设边境MIN

代码

#include<cmath>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
long long r[100001];
long long l[100001];
long long n,m;
long long pos[100001];
long long key[100001];
bool exist[100001];
struct node
{
	long long zhi;
	long long dian;
	bool operator <(const node &t)const
	{
		return zhi>t.zhi;
	}
};
node a[100001];
priority_queue <node> q;

long long read()
{
    long long k=0,f=1;
	char c=getchar();
	while(c<'0'||c>'9') {if(c=='-') f=-1; c=getchar();}
	while(c>='0'&&c<='9') {k=k*10+(c-'0'); c=getchar();}
	return k*f;
}


int main()
{
	freopen("lx.in","r",stdin);
	freopen("lx.out","w",stdout);
	n=read();
	m=read();
	long long i,j,k;
	pos[1]=read();
	for(i=2;i<=n;i++)
	  {
	  	pos[i]=read();
	  	key[i-1]=pos[i]-pos[i-1];
	  }
	for(i=1;i<=n-1;i++)
	  {
	  	l[i]=i-1;
	  	r[i]=i+1;
	  }
	r[n-1]=0;
	l[1]=0;
	key[0]=100000000000000LL;
	node t;
	for(i=1;i<n;i++)
	  {
	  	t.zhi=key[i];
	  	t.dian=i;
	    q.push(t);
	  }
	long long sum=0;
	long long s;
	for(i=1;i<=m;i++)
	  {
	  	while(exist[q.top().dian]) q.pop();
	  	t=q.top();
	    q.pop();
	    s=t.dian;
		sum+=t.zhi;
	    exist[l[s]]=1;
	    exist[r[s]]=1;
	    key[s]=key[l[s]]+key[r[s]]-t.zhi;
	    l[s]=l[l[s]];
	    r[l[s]]=s;
	    r[s]=r[r[s]];
	    l[r[s]]=s;
	    t.zhi=key[s];
	    q.push(t);
	  }
	cout<<sum;
	return 0;
}