bzoj 1150 [CTSC2007]数据备份Backup
bzoj 1150 [CTSC2007]数据备份Backup
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 ≤100 000）表示办公楼的数目，k（1≤ k≤ n/2）表示可利用的网络电缆的数目。接下来的n行每行仅包含一个整数（0≤ s ≤1000 000 000）, 表示每个办公楼到大街起点处的距离。这些整数将按照从小到大的顺序依次出现。
Output
输出应由一个正整数组成，给出将2K个相异的办公楼连成k对所需的网络电缆的最小总长度。
题目大意:要我们在n个数中选择k个数对,让它们的差值之和最小.
这道题我们可以借鉴种树那道题的做法,显然我们要选择相邻的两个数来组成数对,那么我们可以建立一个小根堆,里面存下所有相邻的数的差值,用val表示.但是,如果我们选择了两个建筑,那么这两个建筑以后就不可用了.所以选择一个val时它左边和右边都不能选择了.所以我们取出一个val时记得要补上他左边和右边的和减去当前val的值.这是什么意思呢?就是说我们先选择这个数,如果补上的这个数又出现在堆顶了,那么就相当于选择了它左右两边的数,这样就能完美解决这个问题.至于左右两边的用链表解决就好了.
#include <cstdio>
#include <queue>
#include <algorithm>
using namespace std;
typedef long long LL;
typedef pair<LL,int>pli;
static const int maxm=2e6+10;
static const LL INF=~(1<<31);
priority_queue< pli,vector<pli>,greater<pli> >Q;
int l[maxm],r[maxm];
bool vis[maxm];
LL val[maxm],d[maxm];
int n,k,m,cnt;
LL ans;
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)scanf("%lld",&d[i]);
for(int i=1;i<n;i++){
val[++cnt]=d[i+1]-d[i];
Q.push(make_pair(val[cnt],cnt));
l[cnt]=cnt-1;
r[cnt]=cnt+1;
}
l[cnt+1]=cnt;val[0]=INF;
r[0]=1;val[cnt+1]=INF;
cnt++;
for(int i=1;i<=k;i++){
while(vis[Q.top().second])Q.pop();
pli now=Q.top();Q.pop();
int id=now.second;
ans+=now.first;
int rc=r[id];int lc=l[id];
l[id]=l[lc];r[id]=r[rc];
r[l[id]]=id;l[r[id]]=id;
val[id]=val[rc]+val[lc]-now.first;
vis[lc]=1;vis[rc]=1;
Q.push(make_pair(val[id],id));
}
printf("%lld\n",ans);
return 0;
}
