后悔贪心法 种树（洛谷 P1484）

种树

题目描述
cyrcyr今天在种树，他在一条直线上挖了n个坑。这n个坑都可以种树，但为了保证每一棵树都有充足的养料，cyrcyr不会在相邻的两个坑中种树。而且由于cyrcyr的树种不够，他至多会种k棵树。假设cyrcyr有某种神能力，能预知自己在某个坑种树的获利会是多少（可能为负），请你帮助他计算出他的最大获利。

输入格式

第一行，两个正整数n,k。

第二行，n个正整数，第i个数表示在直线上从左往右数第i个坑种树的获利。

输出格式

输出1个数，表示cyrcyr种树的最大获利。

输入 #1

6 3
100 1 -1 100 1 -1

输出 #1

200

贪心+优先队列；

这种贪心题配上优先队列，多多少少是考一定的思维的，有时候会非常难想；

这道题是可以反悔的贪心，跟这道题有点相似，具体贪心证明洛谷题解上有，不多说了；反正这种题目都大概是先把最多的贪掉，然后记录差值，看是否可以进行反悔操作；

代码：

#include<bits/stdc++.h>
#define LL long long
#define pa pair<int,int>
using namespace std;
struct Node{
	int date,post;
}an[500100];
int l[500100],r[500100],vis[500100];
priority_queue< pa >qu;
LL sum;
int main(){
	int n,k;
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++){
		scanf("%d",&an[i].date);
		an[i].post=i;
		l[i]=i-1;
		r[i]=i+1;
		qu.push(pa(an[i].date,i)); 	
	}
	while(k--){
		while(vis[qu.top().second]) qu.pop();
		pa s=qu.top();
		qu.pop();
		if(s.first<0) break;
		sum+=s.first;
		int p=s.second;
		an[p].date=an[l[p]].date+an[r[p]].date-an[p].date;//差值，方便以后反悔操作
		vis[l[p]]=vis[r[p]]=1;
		l[p]=l[l[p]];r[l[p]]=p;
        r[p]=r[r[p]];l[r[p]]=p; 
        qu.push(pa(an[p].date,p));
	}
	cout<<sum<<endl;
	return 0;
}