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/ +

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