BZOJ1150 [CTSC2007]数据备份Backup 【堆 + 链表】

题目

　　你在一家 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 的网络电缆，满足距离之和最小的要求。

输入格式

第一行包含整数n和k

其中n（2≤n≤100000）表示办公楼的数目，k（1≤k≤n/2）表示可利用的网络电缆的数目。

接下来的n行每行仅包含一个整数（0≤s≤1000000000）,表示每个办公楼到大街起点处的距离。

这些整数将按照从小到大的顺序依次出现。

输出格式

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

输入样例

5 2

输出样例

题解

很容易想到最后选择的每对办公楼一定是相邻的，

因为最后\(ans = \sum Ri - Li\)，R越左越好，L越右越好

这样问题就转化为了：有\(n-1\)个数，选取其中k个，使总和最小，所选数不能相邻

直接贪心选最小是错误的，会发现样例都过不了

因为如果选择了最小的位置x，那么\(x - 1\)和\(x + 1\)两个位置都不能选了

由贪心策略可知，x一定比\(x - 1\)和\(x + 1\)要小，所以三个中只选一个一定是x最优

但如果三个中要选两个，即选\(x - 1\)和\(x + 1\)，如果\(v[x - 1] + v[x + 1] - v[x]\)，比剩余的数都要小，那么显然选择两个更优

这样我们就得到一个算法：

建立一个堆，每次选出最小的数删去并统计答案，同时删去相邻的两个元素合并成一个新元素\(v[new] = v[x - 1] + v[x + 1] - v[x]\)加入堆中

对于三个数的讨论是这样，对于5个数的讨论也类似，所以合并出来的元素被选择时同样也要与相邻的元素合并

所以我们再用链表维护序列，写一个支持删除插入并记录元素位置标号的堆，就可以A了

#include<iostream>

#include<cstdio>

#include<cmath>

#include<cstring>

#include<algorithm>

#define LL long long int

#define Redge(u) for (int k = h[u],to; k; k = ed[k].nxt)

#define REP(i,n) for (int i = 1; i <= (n); i++)

#define BUG(s,n) for (int i = 1; i <= (n); i++) cout<<s[i]<<' '; puts("");

#define ls (u << 1)

#define rs (u << 1 | 1)

#define fa (u >> 1)

using namespace std;

const int maxn = 100005,maxm = 200005,INF = 1000000000;

inline int read(){

	int out = 0,flag = 1; char c = getchar();

	while (c < 48 || c > 57){if (c == '-') flag = -1; c = getchar();}

	while (c >= 48 && c <= 57){out = (out << 3) + (out << 1) + c - 48; c = getchar();}

	return out * flag;

}

LL pre[maxm],val[maxm],post[maxm];

struct node{LL v,u;}H[maxm];

LL hsiz,pos[maxm];

LL n,k,A[maxn],ans;

int pd(int u){

	int small;

	while (true){

		small = u;

		if (ls <= hsiz && H[ls].v < H[small].v) small = ls;

		if (rs <= hsiz && H[rs].v < H[small].v) small = rs;

		if (small != u){

			pos[H[small].u] = u; pos[H[u].u] = small;

			swap(H[small],H[u]);

			u = small;

		}else break;

	}

	return u;

}

void pup(int u){

	while (u > 1 && H[fa].v > H[u].v){

		pos[H[fa].u] = u; pos[H[u].u] = fa;

		swap(H[fa],H[u]);

		u = fa;

	}

}

void ins(int u){

	H[++hsiz] = (node){val[u],u}; pos[u] = hsiz;

	pup(hsiz);

}

void del(int u){

	pos[H[hsiz].u] = u;

	swap(H[u],H[hsiz--]);

	u = pd(u);

	pup(u);

}

int main(){

	n = read(); k = read();

	REP(i,n) A[i] = read();

	sort(A + 1,A + 1 + n);

	for (int i = 1; i < n; i++){

		val[i] = A[i + 1] - A[i];

		if (i - 1) post[i - 1] = i,pre[i] = i - 1;

		ins(i);

	}

	node u;

	int x;

	while (k--){

		u = H[1]; x = u.u;

		ans += u.v;

		if (!pre[x]){

			del(1); del(pos[post[x]]);

			pre[post[post[x]]] = 0;

		}

		else if (!post[x]){

			del(1); del(pos[pre[x]]);

			post[pre[pre[x]]] = 0;

		}

		else {

			int l = pre[x],r = post[x];

			del(pos[l]); del(pos[r]);

			H[1].v = val[x] = val[l] + val[r] - val[x]; pos[x] = 1;

			pd(1);

			post[pre[x] = pre[l]] = x;

			pre[post[x] = post[r]] = x;

		}

	}

	cout << ans << endl;

	return 0;

}