Problem

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 = 2 km。这种配对方案需要总长 4km 的网络电缆,满足距离之和最小的要求。

Input

输入文件的第一行包含整数 n 和 k,其中 n(1≤n≤100 000)表示办公楼的数目,k(1≤k≤n/2)表示可利用的网络电缆的数目。 接下来的 n 行每行仅包含一个整数(0≤s≤1000 000 000), 表示每个办公楼到大街起点处的距离。这些整数将按照从小到大的顺序依次出现。

Output

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

Sample Input

5 2 
1 
3 
4 
6 
12

Sample Output

4

HINT

30%的输入数据满足 n≤20。 60%的输入数据满足 n≤10 000

Solution

Analysis

考虑抽象后的题目

给定n-1个点,选k个不相邻的点使其和最小

考虑贪心

每次选最小的,但这样显然是不优秀的

如下例子:

2 1 2 10 (k=2)

先选1是不优秀的

对于这种情况我们考虑添加一个反悔机制

即当我们选择了a[i]我们重新令a[i]=a[i].lst+a[i].nxt-a[i]即解决问题

Attention

注意赋值inf

Code

//Code by Enderturtle
#include<bits/stdc++.h>
#define rep(i,a,b) for(register int i=a;i<=b;++i)
#define repe(i,a) for(register int i=head[a];i;i=e[i].nxt)
#define il inline
#define pii pair<int,int>
#define mp(a,b) make_pair(a,b)
typedef long long ll;
using namespace std;
il void filejudge(){
    freopen(".in","r",stdin);
    freopen(".out","w",stdout);
}
il int read(){
    int x=0;bool f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=0;ch=getchar();}
    while(isdigit(ch)){x=(x<<3)+(x<<1)+(ch^48);ch=getchar();}
    return f?x:-x;
}
/*----- head end -----*/
const int N=100010;
int n,k;
ll s[N];
struct my_list{
    int lst,nxt;
    ll res;
} a[N];
bool usd[N];
il void del(int id){
    // del id.lst id.nxt
    a[id].lst=a[a[id].lst].lst;
    a[id].nxt=a[a[id].nxt].nxt;
    a[a[id].lst].nxt=id;
    a[a[id].nxt].lst=id;
}
struct node{
    int id;
    ll res;
    bool operator <(const node & other) const{
        return res>other.res;
    }
};
priority_queue<node> q;
ll ans=0;
int main(){
    n=read();k=read();
    rep(i,1,n)
        scanf("%lld",&s[i]);
    rep(i,1,n-1){
        a[i].lst=i-1;
        a[i].nxt=i+1;
        a[i].res=s[i+1]-s[i];
        q.push((node){i,a[i].res});
    }
    a[0].res=a[n].res=LLONG_MAX/3;
    while(k>0 && !q.empty()){
        int id=q.top().id;ll res=q.top().res;q.pop();
        if(usd[id]) continue;
        k--;
        ans+=res;
        usd[a[id].lst]=usd[a[id].nxt]=1;
        a[id].res=a[a[id].lst].res+a[a[id].nxt].res-a[id].res;
        q.push((node){id,a[id].res});
        del(id);
    }
    printf("%lld\n",ans);
    return 0;
}

一个好奇的人