Description
你在一家 IT 公司为大型写字楼或办公楼(offices)的计算机数据做备份。然而数据备份的工作是枯燥乏味
的,因此你想设计一个系统让不同的办公楼彼此之间互相备份,而你则坐在家中尽享计算机游戏的乐趣。已知办公
楼都位于同一条街上。你决定给这些办公楼配对(两个一组)。每一对办公楼可以通过在这两个建筑物之间铺设网
络电缆使得它们可以互相备份。然而,网络电缆的费用很高。当地电信公司仅能为你提供 K 条网络电缆,这意味
着你仅能为 K 对办公楼(或总计2K个办公楼)安排备份。任一个办公楼都属于唯一的配对组(换句话说,这 2K
个办公楼一定是相异的)。此外,电信公司需按网络电缆的长度(公里数)收费。因而,你需要选择这 K 对办公
楼使得电缆的总长度尽可能短。换句话说,你需要选择这 K 对办公楼,使得每一对办公楼之间的距离之和(总距
离)尽可能小。下面给出一个示例,假定你有 5 个客户,其办公楼都在一条街上,如下图所示。这 5 个办公楼分
别位于距离大街起点 1km, 3km, 4km, 6km 和 12km 处。电信公司仅为你提供 K=2 条电缆。
上例中最好的配对方案是将第 1 个和第 2 个办公楼相连,第 3 个和第 4 个办公楼相连。这样可按要求使用
K=2 条电缆。第 1 条电缆的长度是 3km-1km=2km ,第 2 条电缆的长度是 6km-4km=2km。这种配对方案需要总长
4km 的网络电缆,满足距离之和最小的要求。
Input
第一行包含整数n和k
其中n(2≤n≤100000)表示办公楼的数目,k(1≤k≤n/2)表示可利用的网络电缆的数目。
接下来的n行每行仅包含一个整数(0≤s≤1000000000),表示每个办公楼到大街起点处的距离。
这些整数将按照从小到大的顺序依次出现。
Output
输出应由一个正整数组成,给出将2K个相异的办公楼连成k对所需的网络电缆的最小总长度。
Sample Input
5 2
1
3
4
6
12
Sample Output
4
首先我们先算出相邻两栋楼之间的差值,那如果没有不能使一栋楼同时匹配其他楼的限制,那么我们直接贪心从小到大匹配即可。有了这个限制之后怎么做呢?我们先将所有的差值放入一个小根堆,然后我们选取了一组匹配,那么就把它左边和右边的匹配删去。但是删完就好了嘛?我们还要将它左边的值加上右边的值再减去这组匹配的值放入堆中。如果我们下次选取了这个值,就相当于不选当前这组匹配,而选择其两旁的。
我们只要对每一组匹配,都记录它左边的匹配和右边的匹配(会随着删除随时更新),就可以实现。
#include<bits/stdc++.h>
#define mp make_pair
#define pa pair<int,int>
#define inf 0x3f3f3f3f
using namespace std;
int read(){
char c;int x;while(c=getchar(),c<'0'||c>'9');x=c-'0';
while(c=getchar(),c>='0'&&c<='9') x=x*10+c-'0';return x;
}
int n,k,lst,ans,pre[100005],suf[100005],a[100005];
priority_queue<pa,vector<pa>,greater<pa> > q;
int main()
{
n=read();k=read();lst=0;
for(int i=1;i<=n;i++){
int x=read();
a[i]=x-lst;lst=x;
pre[i]=i-1;suf[i]=i+1;
}
pre[2]=0;suf[n]=0;
for(int i=2;i<=n;i++) q.push(mp(a[i],i));
for(int i=1;i<=k;i++){
while(q.top().first!=a[q.top().second]) q.pop(); //删除不匹配的点
pa e=q.top();q.pop();ans+=e.first;
int l=pre[e.second],r=suf[e.second];
a[e.second]=(l&&r)?min(a[l]+a[r]-a[e.second],inf):inf; //更新匹配的值
a[l]=a[r]=inf;
suf[pre[e.second]=pre[l]]=e.second;pre[suf[e.second]=suf[r]]=e.second;
q.push(mp(a[e.second],e.second)); //重新塞入
}
printf("%d",ans);
return 0;
}