算是套路的贪心,因为给出数据是有序的,先将相邻的搞成线段存进堆里,每次取出一个最小的值,加到总值中,并改变左右的链表值,并将左右线段值存入当前位置作为撤销操作。
#include <cstdio>
#include <algorithm>
#include <queue>
#define mk(a,b) make_pair(a,b)
using namespace std;
const int inf=0x3f3f3f3f;
inline void read(int &tmk)
{
tmk=0;char ch=getchar(); int fu=1;
for (;ch<'0'||ch>'9';ch=getchar()) if (ch=='-') fu=-1;
for (;ch>='0'&&ch<='9';ch=getchar()) tmk=tmk*10+ch-'0';tmk*=fu;
}
const int N=100005;
typedef pair<int,int> pii;
priority_queue<pii,vector<pii>,greater<pii> > q;
int n,m;
int e[N],pre[N],nxt[N];
int main()
{
register int i,j;
read(n),read(m);
int x=0,y;
for (i=1;i<=n;i++)
{
read(y);
e[i]=y-x;pre[i]=i-1,nxt[i]=i+1;x=y;
}
int ans=0;
for (i=2;i<=n;i++) q.push(mk(e[i],i));
pre[2]=0,nxt[n]=0;
for (i=1;i<=m;i++)
{
while (q.top().first!=e[q.top().second]) q.pop();
int x=q.top().second,l=pre[x],r=nxt[x];
ans+=e[x]; q.pop();
pre[nxt[x]=nxt[r]]=x;
nxt[pre[x]=pre[l]]=x;
e[x]=l&&r?min(inf,e[l]+e[r]-e[x]):inf;
e[l]=e[r]=inf;
q.push(mk(e[x],x));
}
printf("%d\n",ans);
return 0;
}