LINK:数据备份
以前做过这种贪心 不过没有好好的证明 这次来严格的证明一下。
不难发现 最后的答案 选择的所有两对公司必然相邻。
所以排序后 把数组变成ai-ai-1. 这样问他的模型就是 n-1个数从中选出k个数 且任意两个数不能相邻 求和的最小值。
k==1时显然是全局最小值。
k==2的时候 有两种方法:全局最小值 和全局最小值不相邻的 最小数字。还有一种 把全局最小值扔了 选他们两边的数字。
此时 选出全局最小值之后把左右两边元素给去掉 因为这两个决策假了。
加入一个新决策 :两边元素之和-中间的元素。
然后可以发现 此时决策集合完全覆盖了我们上述的分析。
考虑拓展到k比较大的情况。
我们不断维护决策集合的完整性。而且对于每一个局面都是一样的。所以从递归来看这是正确的。
换个角度:从一个局面最优转到另一个局面 可以发现 此时如果我们选择了上次新决策那么符合我们前面的结论2.
如果选择了其他元素符合前面的结论1.
对于一个新局面来说其最优也是来自两个结论。
综上 做出的所有决策都是 结论1和结论2的一种 这对于所有的局面都是最优的选择 所以是正确的。
const int MAXN=100010;
int n,k;
ll ans;
int l[MAXN],r[MAXN];
ll w[MAXN];
ll a[MAXN];
multiset<pii>s;
multiset<pii>:: iterator it,itt,itt1;
int main()
{
//freopen("1.in","r",stdin);
get(n);get(k);
rep(1,n,i)get(a[i]);
sort(a+1,a+1+n);
rep(1,n-1,i)w[i]=a[i+1]-a[i],s.insert(mk(w[i],i)),l[i]=i-1,r[i]=i+1;
w[0]=inf;w[n]=inf;s.insert(mk(w[0],0));s.insert(mk(w[n],n));
while(k--)
{
it=s.begin();
ans+=(*it).F;
int x=(*it).S;
int L=l[x];
int R=r[x];
s.erase(it);
itt=s.find(mk(w[L],L));
s.erase(itt);
itt1=s.find(mk(w[R],R));
s.erase(itt1);
w[x]=w[R]+w[L]-w[x];
l[x]=l[L];r[x]=r[R];
r[l[L]]=x;l[r[R]]=x;
s.insert(mk(w[x],x));
}
putl(ans);
return 0;
}