BZOJ 1150--数据备份(链表&堆&贪心)

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对所需的网络电缆的最小总长度。

Sample Input

Sample Output

Solution

　　　　显然最优解一定满足每条电缆都要连接两个相邻的办公楼，否则会有更优的解。。。

　　　　假如对于一条已知的最短的边，但如果要求连接两条电缆。。。

　　　　就会存在最短边的两边恰好是最优解的状态。。。所以要用链表维护一下。。。

#include<cstdio> #include<cstring> #include<cmath> #include<algorithm> #include<iostream> #include<vector> #define LL long long #define maxn 100005 #define inf 2000000000 using namespace std; int n,k; priority_queue< pair<int,int>, vector< pair<int ,int> > ,greater< pair<int,int> > >q; int l[maxn],r[maxn],t[maxn],s[maxn]; bool mark[maxn]; void del(int x){ mark[x]=1; int left=l[x],right=r[x]; l[x]=r[x]=0; r[left]=right; l[right]=left; } int main(){ LL ans=0; scanf("%d%d",&n,&k); for(int i=1;i<=n;i++){ scanf("%d",&s[i]); l[i]=i-1; r[i]=i+1; } t[1]=inf;t[n+1]=inf; q.push(make_pair(t[1],1)); q.push(make_pair(t[n+1],n+1)); for(int i=2;i<=n;i++){ t[i]=s[i]-s[i-1]; q.push(make_pair(t[i],i)); } for(int i=1;i<=k;i++){ while(mark[q.top().second])q.pop(); int x=q.top().second; ans+=t[x]; q.pop(); t[x]=t[l[x]]+t[r[x]]-t[x]; del(l[x]); del(r[x]); q.push(make_pair(t[x],x)); } printf("%lld\n",ans); return 0; }