【BZOJ 1150】 [CTSC2007]数据备份Backup|链表|堆|贪心

同 2151

不过不是环 需要在两端加上最大数 也就是 pre[2]=nxt[n]=0; num[0]=INF;

#include <queue>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

const int MAXN=200000+10;
int pre[MAXN],nxt[MAXN],num[MAXN];
bool b[MAXN];
struct H
{
	int id;
	int v;
	bool operator < (const H b) const
	{
		return v>b.v;
	}
} ;
int n,m,ans;
priority_queue <H> q;
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) scanf("%d",&num[i]);
	for(int i=n;i>=2;i--)
	{
		num[i]-=num[i-1];
		q.push((H){i,num[i]});
		pre[i]=i-1;
		nxt[i]=i+1;
	}
	num[0]=2000000000;
	pre[2]=0;
	nxt[n]=0;
	while(m--)	
	{
		while(b[q.top().id]) q.pop();
		H tmp=q.top();//cout<<tmp.id<<' '<<tmp.v<<endl;
		q.pop();
		ans+=tmp.v;
		b[pre[tmp.id]]=b[nxt[tmp.id]]=true;
		num[tmp.id]=-num[tmp.id]+num[pre[tmp.id]]+num[nxt[tmp.id]];
		tmp.v=num[tmp.id];
		
		pre[tmp.id]=pre[pre[tmp.id]];
		nxt[tmp.id]=nxt[nxt[tmp.id]];
		nxt[pre[tmp.id]]=tmp.id;
		pre[nxt[tmp.id]]=tmp.id;
		q.push(tmp);
	}
	cout<<ans<<endl;
	return 0;
}