BZOJ1150：[APIO/CTSC2007]数据备份——题解

https://www.lydsy.com/JudgeOnline/problem.php?id=1150

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

一道需要些简单思（trick）维的题。

发现了我们只能取相邻两个办公楼之后我们就能够写出dp了，就不说了。

正解考虑贪心去取：我们将每一对办公楼的距离加入到堆里面，然后每次弹出最小，并且弹出的两侧办公楼对就不能再取了。是否可以呢？

显然是不行的，你会发现样例就是个反例：2 1 2 6，你取完1之后只能取6，结果输出了7。

于是我们要给程序一个“后悔“的机会：将弹出的两侧办公楼对合为一对，其距离为两个办公楼对的距离和-弹出的办公楼距离。

这样我们取这个对的时候就相当于减去了中间的对而取了旁边两侧的对，就是一个“后悔”的过程啦！

维护两侧的对可以用链表来实现。

#include<map>
#include<cmath>
#include<stack>
#include<queue>
#include<cstdio>
#include<cctype>
#include<vector>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
#define fi first
#define se second
const int N=1e5+5;
const int INF=1e9+1;
inline int read(){
    int X=0,w=0;char ch=0;
    while(!isdigit(ch)){w|=ch=='-';ch=getchar();}
    while(isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
    return w?-X:X;
}
map<pii,bool>mp;
priority_queue<pii,vector<pii>,greater<pii> >q;
int n,k,a[N],pre[N],nxt[N];
int main(){
    n=read(),k=read();
    for(int i=1;i<=n;i++)a[i]=read();
    for(int i=1;i<n;i++){
        a[i]=a[i+1]-a[i];
        pre[i]=i-1;nxt[i-1]=i;
        q.push(pii(a[i],i));
    }
    nxt[n-1]=n;pre[0]=0;nxt[n]=n;a[0]=a[n]=INF;
    int ans=0;
    while(k--){
        pii p=q.top();q.pop();
        if(mp.count(p)){k++;mp.erase(p);continue;}
        ans+=p.fi;
        a[p.se]=a[pre[p.se]]+a[nxt[p.se]]-p.fi;
        mp[pii(a[pre[p.se]],pre[p.se])]=1;
        mp[pii(a[nxt[p.se]],nxt[p.se])]=1;
        q.push(pii(a[p.se],p.se));
        pre[p.se]=pre[pre[p.se]];
        nxt[p.se]=nxt[nxt[p.se]];
        nxt[pre[p.se]]=p.se;
        pre[nxt[p.se]]=p.se;
    }
    printf("%d\n",ans);
    return 0;
}

+++++++++++++++++++++++++++++++++++++++++++

+本文作者：luyouqi233。 　　　　　　　　　　　　　　+

+欢迎访问我的博客：http://www.cnblogs.com/luyouqi233/ +

+++++++++++++++++++++++++++++++++++++++++++