【题目】
你在一家 IT 公司为大型写字楼或办公楼\((offices)\)的计算机数据做备份。然而数据备份的工作是枯燥乏味的,因此你想设计一个系统让不同的办公楼彼此之间互相备份,而你则坐在家中尽享计算机游戏的乐趣。
已知办公楼都位于同一条街上。你决定给这些办公楼配对(两个一组)。每一对办公楼可以通过在这两个建筑物之间铺设网络电缆使得它们可以互相备份。
然而,网络电缆的费用很高。当地电信公司仅能为你提供 \(K\) 条网络电缆,这意味着你仅能为 \(K\) 对办公楼(或总计 \(2K\) 个办公楼)安排备份。任一个办公楼都属于唯一的配对组(换句话说,这 \(2K\) 个办公楼一定是相异的)。
此外,电信公司需按网络电缆的长度(公里数)收费。因而,你需要选择这 \(K\) 对办公楼使得电缆的总长度尽可能短。换句话说,你需要选择这 \(K\) 对办公楼,使得每一对办公楼之间的距离之和(总距离)尽可能小。
下面给出一个示例,假定你有 \(5\) 个客户,其办公楼都在一条街上,如下图所示。这 \(5\) 个办公楼分别位于距离大街起点 \(1km, 3km, 4km, 6km\) 和 \(12km\) 处。电信公司仅为你提供 \(K=2\) 条电缆。
输入格式:
输入文件的第一行包含整数 \(n\) 和 \(k\),其中 \(n(1≤n≤100 000)\)表示办公楼的数目,\(k(1≤k≤n/2)\)表示可利用的网络电缆的数目。
接下来的 \(n\) 行每行仅包含一个整数\(s(0≤s≤1000 000 000)\), 表示每个办公楼到大街起点处的距离。这些整数将按照从小到大的顺序依次出现。
输出格式:
输出文件应当由一个正整数组成,给出将 \(2K\) 个相异的办公楼连成 \(K\) 对所需的网络电缆的最小总长度。
【Input】
5 2
1
3
4
6
12
【Output】
4
【分析】
要使每根线的长度尽量小,则每对线的办公楼一定是相邻的,所以我们可以建立一个差分数组\(d[i]\)来储存相邻办公楼的距离。
于是,问题转化为:在\(d[i]\)中找出\(k\)个不相邻的数,使它们的和最小。
考虑建一个小根堆,并将\(d[i]\)中相邻的数用链表储存,将堆和链表一一对应,然后进行\(k\)次操作,每次取堆顶最小值,再将其用\(d[i+1]+d[i-1]-d[i]\)代替,最后将\(d[i],d[i+1],d[i-1]\)合并为一堆即可。
至于为什么可以将\(d[i]\)用\(d[i+1]+d[i-1]-d[i]\)代替,考虑到若\(d[i]\)不是最优,则为了保证结果最小,那么必选\(d[i-1]\)与\(d[i+1]\),这样一来就需排除\(d[i]\)与\(d[i+2],d[i-2]\),所以这样做的话无论\(d[i]\)是否为最优解都能找到正确答案。
【代码】
#include<cstdio>
#include<cstring>
#include<queue>
#define maxn 100010
#define INF 1e9
using namespace std;
struct offices{
int id,len;
bool operator < (const offices& a)const{return a.len<len;}
}tmp;
int n,k,ans=0;
int s[maxn],last[maxn],nxt[maxn],d[maxn];
priority_queue<offices>q;
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++) scanf("%d",&s[i]);
for(int i=1;i<n;i++){
d[i]=s[i+1]-s[i];last[i]=i-1;nxt[i]=i+1;//链表存储
q.push((offices){i,d[i]});//入堆
}
nxt[n-1]=0;//防止越界
while(k--){
while(q.top().len!=d[q.top().id]) q.pop();//已经合并过
tmp=q.top();q.pop();
int l=last[tmp.id],r=nxt[tmp.id];
ans+=tmp.len;
last[nxt[r]]=tmp.id;nxt[last[l]]=tmp.id;//合并为一堆
nxt[tmp.id]=nxt[r];last[tmp.id]=last[l];
if(l&&r) d[tmp.id]=d[l]+d[r]-d[tmp.id];
else d[tmp.id]=INF;
d[l]=d[r]=INF;
q.push((offices){tmp.id,d[tmp.id]});//将d[l]+d[r]-d[i]入堆
}
printf("%d",ans);
return 0;
}