持续创作，加速成长！这是我参与「掘金日新计划 · 6 月更文挑战」的第14天，点击查看活动详情

[APIO/CTSC 2007] 数据备份

题目描述

你在一家 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 = 2 km。这种配对方案需要总长 4km 的网络电缆，满足距离之和最小的要求。

输入格式

输入文件的第一行包含整数 n 和 k，其中 n(1≤n≤100 000)表示办公楼的数目，k(1≤k≤n/2)表示可利用的网络电缆的数目。

接下来的 n 行每行仅包含一个整数(0≤s≤1000 000 000), 表示每个办公楼到大街起点处的距离。这些整数将按照从小到大的顺序依次出现。

输出格式

输出文件应当由一个正整数组成，给出将 2K 个相异的办公楼连成 K 对所需的网络电缆的最小总长度。

样例 #1

样例输入 #1

样例输出 #1

提示

30%的输入数据满足 n≤20。

60%的输入数据满足 n≤10 000

反悔贪心

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#define inf 0x3f3f3f3f
#define N 100005
#define rep(i,x,y) for (register int i=x;i<=y;++i) 
using namespace std; 
int n,k,list,ans,pre[N],suf[N],a[N],t,T; 
priority_queue<pair< int ,int >,vector< pair < int ,int > >,greater<pair< int ,int > > > q; 
int main(){ 
	scanf("%d%d",&n,&k); 	
	ans=0; list=0;
	memset(pre,0,sizeof(pre)); 
	memset(suf,0,sizeof(suf));
	while (q.size()) q.pop(); 
	rep(i,1,n) {
		scanf("%d",&t); a[i]=t-list; list=t; 
		pre[i]=i-1,suf[i]=i+1;  
	}
	pre[2]=suf[n]=0; 
	for(register int i=2;i<=n;++i) q.push(make_pair(a[i],i)); 
	for(register int i=1;i<=k;++i) {
		while (q.top().first!=a[q.top().second]) q.pop(); 
		pair< int ,int > e=q.top(); q.pop(); ans+=e.first; 
		int l=pre[e.second],r=suf[e.second]; 
		a[e.second]=(l&&r)?min(a[l]+a[r]-a[e.second],inf):inf; a[l]=a[r]=inf; 
		suf[pre[e.second]=pre[l]]=e.second,pre[suf[e.second]=suf[r]]=e.second;
		q.push(make_pair(a[e.second],e.second)); 
	}
	printf("%d\n",ans); 
	return 0; 
}