BZOJ1150: [CTSC2007]数据备份Backup

1150: [CTSC2007]数据备份Backup

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 682  Solved: 294
[Submit][Status]

Description

 

Input

输 入的第一行包含整数n和k，其中n（2 ≤ 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

上面的样例输入给出了前面描述的示例情形 对于每一个测试点，如果写到输出文件中的答案正确，则得到该测试点100%的分数，否则得零分。30%的输入数据满足n≤20。60%的输入数据满足n≤10 000。

Source

 题解：

原来这题是CTSC的原题啊，我说为什么zxr的那场比赛有那么多人能想到标算，orz。

题解戳以前的链接http://www.cnblogs.com/zyfzyf/p/3988776.html

代码：

#include<cstdio>

#include<cstdlib>

#include<cmath>

#include<cstring>

#include<algorithm>

#include<iostream>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<string>

#define inf 1000000000

#define maxn 100000+5

#define maxm 500+100

#define eps 1e-10

#define ll long long

#define pa pair<int,int>

#define for0(i,n) for(int i=0;i<=(n);i++)

#define for1(i,n) for(int i=1;i<=(n);i++)

#define for2(i,x,y) for(int i=(x);i<=(y);i++)

#define for3(i,x,y) for(int i=(x);i>=(y);i--)

#define mod 1000000007

using namespace std;

inline int read()

{

    int x=0,f=1;char ch=getchar();

    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}

    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}

    return x*f;

}
int n,k,pre[maxn],nxt[maxn],len[maxn];
priority_queue<pa,vector<pa>,greater<pa> >q;

int main()

{

    freopen("input.txt","r",stdin);

    freopen("output.txt","w",stdout);

    n=read();k=read();int x=0,y;
    for1(i,n)
    {
        y=read();
        len[i]=y-x;x=y;pre[i]=i-1;nxt[i]=i+1;
    }
    pre[2]=0;nxt[n]=0;int ans=0;//只有n-1个距离才能计入权值
    for2(i,2,n)q.push(pa(len[i],i));
    for1(i,k)
    {
        while(len[q.top().second]!=q.top().first)q.pop();//已被删去的点
        int c=q.top().second,l=pre[c],r=nxt[c];
        ans+=len[c];q.pop();
        pre[nxt[c]=nxt[r]]=nxt[pre[c]=pre[l]]=c;//删去c的前趋后继
        len[c]=l&&r?min(inf,len[l]+len[r]-len[c]):inf;//加入一个新点，反悔点
        len[l]=len[r]=inf;//删去点的权值记为inf，使它不会被堆弹出
        q.push(pa(len[c],c));//加入
    }
    printf("%d\n",ans);

    return 0;

}