bzoj1150-[CTSC2007]数据备份Backup

Posted 2021-02-10 ubospica

Description

你在一家 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
1
3
4
6
12

样例输出

4

Solution

贪心反悔.

显然题目等价于求最小的(k)个间隔, 且两两不相邻.

如果没有第二个约束条件, 可以放入堆中, 求最小的 (k) 个即可.

如果有第二个约束条件, 考虑连续的三个 ((a, b, c)), 要增加一个选择. 如果都没选, 可以选中 (b); 如果选中了 (b), 可以选择 (a,c), 同时取消 (b). 显然所有的选择方案都可以看做这两种方法的组合.

因此维护一个堆, 每次选中最小值b时再加入 (v_c + v_a - v_b), 选(k)次即可.

然后考虑端点, 可以证明如果选中了端点, 那么与端点相邻的点一定不选. 需要特判一下.

另外, 似乎也可以用模拟费用流来得到上面的做法:

【bzoj1150】[CTSC2007]数据备份Backup 模拟费用流+链表+堆 - GXZlegend - 博客园

Code

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
#include<queue>
#include<set>
#include<map>
using namespace std;
#define rep(i,l,r) for(register int i=(l);i<=(r);++i)
#define repdo(i,l,r) for(register int i=(l);i>=(r);--i)
#define il inline
typedef double db;
typedef long long ll;

//---------------------------------------
const int nsz=1e5+50,ninf=(int)1e9+500;
int n,k,line[nsz],ans=0;

struct tnd{int p,v;};
bool operator<(tnd a,tnd b){return a.v>b.v;}
priority_queue<tnd> pq;

int pr[nsz],ne[nsz];
int sol(){
    tnd tmp;
    rep(i,1,k){
        tmp=pq.top(),pq.pop();
        while(line[tmp.p]!=tmp.v)tmp=pq.top(),pq.pop();
        ans+=tmp.v;
        int p=tmp.p,l=pr[p],r=ne[p];
//      if(l==0||r==0)continue;
        line[p]=((l&&r)?min(line[l]+line[r]-line[p],ninf):ninf);
        line[l]=line[r]=ninf;
        pr[p]=pr[l],ne[pr[p]]=p,ne[p]=ne[r],pr[ne[p]]=p;
        pq.push((tnd){p,line[p]});
    }
    return ans;
}

int main(){
    ios::sync_with_stdio(0),cin.tie(0);
    cin>>n>>k;
    rep(i,1,n)cin>>line[i];
    repdo(i,n,2)line[i]-=line[i-1],pq.push((tnd){i,line[i]}),pr[i]=i-1,ne[i]=i+1;
    pr[2]=ne[n]=0,line[1]=line[n+1]=ninf;
    cout<<sol()<<'
';
    return 0;
}