Codeforces 958E2 Guard Duty (medium) 反悔贪心

一道经典反悔贪心模板题,但是 $k$只有 $5000$,所以题解变成了 $k^2$的优化 $dp$.

题意

你的宇宙飞船要会见$k$位领导人,只有一些固定时间能让领导人上下车,但是值得高兴的是领导人只会待一会.你不能同时会见两个以上的领导人,同一个时间只能让一个领导人上车或者下车.问会见这些领导人至少要多少时间.

题解

转化模型.对时间排序后计算相邻差,则问题转化为取$k$个数使得没有两个相邻,此为经典反悔贪心模板.
首先构建双向链表,取走最大的数的时候,由于可能存在两边之和大于它本身的情况,我们删去两边的结果,将$a_{i+1}+a_{i-1}-a_i$赋值给$a_i$,这样如果再取最大值,取到$a_i$的时候,答案就相当于取了两边的数字.
这样的贪心进行$k$次便可得到最优解.
谢谢大家.

const int yuzu=5e5;
const ll inf=1e18;
typedef ll fuko[yuzu|10];
typedef pair<ll,int> pli;
fuko net,pre,a;
int main() {
  int n,k,i;
  read(k),read(n);
  for (i=1;i<=n;++i) read(a[i]);
  sort(a+1,a+n+1);
  for (i=1;i<n;++i) {
    a[i]=a[i+1]-a[i];
    net[i]=i+1,pre[i]=i-1;
  } 
  net[n-1]=0;
  ll zw=0;
  priority_queue<pli,vector<pli>,greater<pli>> q;
  for (i=1;i<n;++i) q.push(pli(a[i],i));
  for (;k--;) {
    for (;q.top().first^a[q.top().second];) q.pop();
    int u=q.top().second,l=pre[u],r=net[u];
    zw+=a[u]; q.pop();
    pre[net[u]=net[r]]=u;
    net[pre[u]=pre[l]]=u;
    a[u]=l&&r?min(inf,a[l]+a[r]-a[u]):inf;
    a[l]=a[r]=inf;
    q.push(pli(a[u],u));
  }
  printf("%lld\n",zw);
}