【BZOJ 2151】 种树|堆|贪心|链表

用堆优化 链表记录 每次选最大的

然后 用 num[pre]+num[nxt]-num[now] 替换num[now] 同时更新链表

具体见 http://www.myexception.cn/program/1967776.html

#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;
	if((n/2)<m)
	{
		cout<<"Error!"<<endl;
		return 0;
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&num[i]);
		q.push((H){i,num[i]});
		pre[i]=i-1;
		nxt[i]=i+1;
	}
	pre[1]=n;
	nxt[n]=1;
	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;
}
/*
10 4
10 8 5 2 3 6 9 7 4 1
*/