洛谷P1484 种树（反悔贪心，双向链表+堆）

https://www.luogu.com.cn/problem/P1484

最近这个“反悔贪心”出现频率有点儿高。。。

如果选2堆：
如果(a[x])是最大的，那么(a[x-1])和(a[x+1])要么都不选，要么扔掉(a[x])之后同时选。
证明：
假设没选最大的(a[x])，选了(a[x-1])而没选(a[x+1])
设选的另一个是(a[k])，(a[k])和(a[x])一定不相邻，选(a[x])和(a[k])一定更优

所以就可以选出最大的(a[x])之后，删除掉(a[x-1])和(a[x+1])，然后修改(a[x])为(a[x-1]+a[x+1]-a[x])，这样当第二次选择(a[x])时，即表示不选(a[x])而改选(a[x-1])和(a[x+1]),也就是实现反悔的过程。

用双向链表和堆实现

#include<cstdio>
#include<queue>
#include<iostream>

using namespace std;

#define N 500002

long long a[N];
struct node
{
	int g;
	bool operator < (node p) const
	{
		return a[g]<a[p.g];
	}
}e[N];
priority_queue<node>q;

bool tra[N];
int l[N],r[N];

int main()
{
//	freopen("data.txt","r",stdin);
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i)  scanf("%lld",&a[i]);
	for(int i=2;i<=n;++i) l[i]=i-1;
	for(int i=1;i<n;++i) r[i]=i+1;
	node now;
	for(int i=1;i<=n;++i) 
	{
		now.g=i;	
		q.push(now);
	}
	long long ans=0;
	while(k && !q.empty())
	{
		now=q.top();
		q.pop();
		if(a[now.g]<=0) break;
		if(tra[now.g]) continue;
		ans+=a[now.g];		
		--k;
		tra[l[now.g]]=true;
		tra[r[now.g]]=true;
		a[now.g]=a[l[now.g]]+a[r[now.g]]-a[now.g];
		l[now.g]=l[l[now.g]];
		r[l[now.g]]=now.g;
		r[now.g]=r[r[now.g]];
		l[r[now.g]]=now.g;
		q.push(now);
	}
	printf("%lld",ans);
}