147. 数据备份

题目链接

147. 数据备份

你在一家 IT 公司为大型写字楼或办公楼的计算机数据做备份。

然而数据备份的工作是枯燥乏味的，因此你想设计一个系统让不同的办公楼彼此之间互相备份，而你则坐在家中尽享计算机游戏的乐趣。

已知办公楼都位于同一条街上，你决定给这些办公楼配对（两个一组）。

每一对办公楼可以通过在这两个建筑物之间铺设网络电缆使得它们可以互相备份。

然而，网络电缆的费用很高。

当地电信公司仅能为你提供 \(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\) 表示办公楼的数目，\(K\) 表示可利用的网络电缆的数目。

接下来的 \(n\) 行每行仅包含一个整数 \(s\)，表示每个办公楼到大街起点处的距离。

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

输出格式

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

数据范围

\(2 \le n \le 100000\),
\(1 \le K \le n/2\),
\(0 \le s \le 1000000000\)

输入样例：

5 2 
1
3
4
6
12

输出样例：

4

解题思路

反悔贪心，双链表

显然有一条性质：这 \(k\) 对肯定是相邻的
将相邻的值求出，记作 \(d_i\)，考虑选择最小值 \(d_{i}\)，则 \(d_{i-1}\) 或 \(d{i+1}\) 不能选，相反，如果不选择 \(d_i\)，则 \(d_{i-1}\) 和 \(d_{i+1}\) 都必须要选，因为如果其中有一个没选，则都不如选择最小值 \(d_i\) 更优，故只有两种方案，可以考虑用优先队列将这两种方案结合起来，即每次都选择最小值，然后插入 \(d_{i-1}+d_{i+1}-d_i\)，如果选到该值，表示不选 \(d_{i}\) 选择 \(d_{i-1}\) 和 \(d_{i+1}\) 更优，用 \(set\) 搭配双链表实现即可

• 时间复杂度：\(O(nlogn)\)

代码

// Problem: 数据备份
// Contest: AcWing
// URL: https://www.acwing.com/problem/content/149/
// Memory Limit: 64 MB
// Time Limit: 1000 ms
// 
// Powered by CP Editor (https://cpeditor.org)

// %%%Skyqwq
#include <bits/stdc++.h>
 
//#define int long long
#define help {cin.tie(NULL); cout.tie(NULL);}
#define pb push_back
#define fi first
#define se second
#define mkp make_pair
using namespace std;
 
typedef long long LL;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;
 
template <typename T> bool chkMax(T &x, T y) { return (y > x) ? x = y, 1 : 0; }
template <typename T> bool chkMin(T &x, T y) { return (y < x) ? x = y, 1 : 0; }
 
template <typename T> void inline read(T &x) {
    int f = 1; x = 0; char s = getchar();
    while (s < '0' || s > '9') { if (s == '-') f = -1; s = getchar(); }
    while (s <= '9' && s >= '0') x = x * 10 + (s ^ 48), s = getchar();
    x *= f;
}

const int N=1e5+5;
typedef pair<LL,int> PLI;
int n,k,L[N],R[N];
LL a[N];
set<PLI> s;
void del(int x)
{
	R[L[x]]=R[x],L[R[x]]=L[x];
}
int main()
{
    cin>>n>>k;
    for(int i=0;i<n;i++)cin>>a[i];
    for(int i=n-1;i>=1;i--)a[i]-=a[i-1];
    for(int i=0;i<=n;i++)
    {
    	L[i]=i-1,R[i]=i+1;
    	if(i>0&&i<n)s.insert({a[i],i});
    }
    a[0]=a[n]=1e12;
    LL res=0;
    while(k--)
    {
    	auto t=s.begin();
    	s.erase(t);
    	int x=(*t).se;
		int l=L[x],r=R[x];
		s.erase({a[l],l}),s.erase({a[r],r});
		res+=a[x];
		a[x]=a[l]+a[r]-a[x];  
		s.insert({a[x],x});
		del(l),del(r);  	
    }
    cout<<res;
    return 0;
}