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
5 2
1
3
4
6
12
Sample Output
4
数据范围很大,考虑贪心求解
显然最终选出的每对中的两个点必须相邻,否则必定可以找到更优
的解
如果仅仅每次找距离最近的两个点会出现错误
n个点组成 n?1 段,为了方便,在首尾各加 1 段,长度为 +∞, 因此共 n+ 1 段
用双向链表将这 n+ 1 段串起来,并将其加入到堆中 每次选择堆中最小的一段,ans 加上这一段的长度,并将其左右结 点从链表中删除
选出的这一段不删除,修正该段 leni 的长度为 lenlefti +lenrighti?leni 这个思想类似于网络流找增广路的思想,网络流中的退流操作就是 这里的修正过程
注意
39,46行更新操作十分重要,不可忽略。
赋极大值不能是INT_MAX
#include<iostream>
#include<cstdio>
#include<queue>
using namespace std;
const int mx=1e9;
struct xd
{int num,len,pre,nxt;
}a[100005],tmp;
bool operator < (const xd &c,const xd &d)
{return c.len>d.len;
}
priority_queue<xd>q;
int n,k,ans,cur,lst;
bool del[100005];
int main()
{scanf("%d%d",&n,&k);for(int i=1;i<=n;i++){scanf("%d",&cur);a[i].num=i;a[i].pre=i-1;a[i].nxt=i+1;a[i].len=cur-lst;lst=cur;}a[1].len=mx;a[n+1].len=mx;a[n+1].pre=n;for(int i=1;i<=n+1;i++)q.push(a[i]);while(k--){while(del[q.top().num])q.pop();tmp=q.top();tmp=a[tmp.num];q.pop();ans+=tmp.len;del[tmp.pre]=del[tmp.nxt]=1;tmp.len=min(mx,a[tmp.pre].len+a[tmp.nxt].len-tmp.len);tmp.pre=a[tmp.pre].pre;tmp.nxt=a[tmp.nxt].nxt;a[tmp.num]=tmp;a[tmp.nxt].pre=tmp.num;a[tmp.pre].nxt=tmp.num;q.push(tmp);}printf("%d\n",ans);return 0;
}