bzoj 1150: [CTSC2007]数据备份Backup

/**************************************************************
    Problem: 1150
    User: lxy8584099
    Language: C++
    Result: Accepted
    Time:464 ms
    Memory:4004 kb
****************************************************************/
 
/*
    这是什么方法不清楚
    带返回的贪心？
    首先是 线不可能跨过某个办公楼 
    处理出来 n-1个办公楼之间的距离
    优先队列 
    利用链表 每选择了一个距离后就把他左右缩掉 
    因为如果不选这个距离更优  则一定选择了他左右两边的距离
    所以加进 dis[l]+dis[r]-dis[i] 表示一个反悔吧 
*/
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+50;
struct pp {int i,d;};
bool operator < (const pp&a,const pp&b) {return a.d>b.d;} 
int nxt[N],pre[N],n,k,a[N],la,res;
priority_queue< pp > q;
int main()
{
    scanf("%d%d%d",&n,&k,&la); a[0]=a[n]=1e9;// 大一点就不会取到这里了 
    for(int i=1,x;i<n;i++)
        scanf("%d",&x),a[i]=x-la,la=x;
    for(int i=1;i<n;i++) 
        q.push((pp){i,a[i]}),pre[i]=i-1,nxt[i]=i+1;;
    nxt[n]=0,pre[0]=1,pre[n]=n-1;
    while(k--)
    {
        int i,d;
        while(1)
        {
            i=q.top().i;d=q.top().d;q.pop();
            if(a[i]!=d) continue; break;
        }
        res+=d; a[i]=a[pre[i]]+a[nxt[i]]-a[i];
        a[pre[i]]=a[nxt[i]]=0; // 归零 说明已经不能选了 
        nxt[pre[i]=pre[pre[i]]]=pre[nxt[i]=nxt[nxt[i]]]=i;
        q.push((pp){i,a[i]});
    }
    printf("%d\n",res);
    return 0;
}