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对所需的网络电缆的最小总长度。
首先考虑一种错误的贪心,用堆维护,每次选择最短的区间。显然不一定得到最优解,因为取一个最优解可能会失去旁边的两个两个次优解。
如何修正呢?每次取完之后不直接删除,而是删除旁边的两个,把len[left]+len[right]-len[now]插入原来的位置。这样如果之后选了这个元素,就相当于本来没选中间元素,而选了旁边两个元素。
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
#define LL long long
const int oo=2e9+100;
bool del[500010];
int next[300010],pre[300010],val[300010];
struct inter
{
int x,num;
bool operator < (const inter & ii) const
{
return x>ii.x;
}
}i1,i2;
priority_queue<inter> que;
int main()
{
int i,j,k,m,n,p,q,x,y,z,last;
LL ans;
scanf("%d%d",&n,&m);
scanf("%d",&last);
val[1]=oo;
for (i=2;i<=n;i++)
{
scanf("%d",&x);
val[i]=x-last;
last=x;
}
val[n+1]=oo;
for (i=1;i<=n+1;i++)
{
next[i]=i+1;
pre[i]=i-1;
que.push((inter){val[i],i});
}
ans=0;
while (m--)
{
while (1)
{
i1=que.top();
que.pop();
if (!del[i1.num]) break;
}
ans+=i1.x;
x=pre[i1.num];
y=next[i1.num];
next[pre[x]]=i1.num;
pre[i1.num]=pre[x];
pre[next[y]]=i1.num;
next[i1.num]=next[y];
del[x]=del[y]=1;
i1.x=val[i1.num]=val[x]+val[y]-i1.x;
que.push((inter){i1.x,i1.num});
}
printf("%lld\n",ans);
}