洛谷 p1484

题目描述

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

输入输出格式

输入格式：

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

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

输出格式：

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

思路:
k=1时，显然取n个数中取最大的即可（暂不考虑全负的情况）。设最大的数是a[i]。
k=2时，则有两种可能：1、另取一个与a[i]不相邻的a[j]。2、取a[i-1]和a[i+1]。
我们可以发现：如果k=1时最优解为a[i]，那么我们便可以把a[i-1]和a[i+1]进行合并，
因为它们要么同时被选，要么同时落选（证明不难，请自行解决）。而且，我们还注意到：
当选了a[i-1]和a[i+1]时，获利便增加了a[i-1]+a[i+1]-a[i]。所以当a[i]被选时，我们
就可以删去a[i-1]和a[i+1]，并把a[i]改成a[i-1]+a[i+1]-a[i]，重新找最大的。

代码:

#include<bits/stdc++.h>
using namespace std;
#define ll long long 
const int maxn=5e5+10;
int l[maxn],r[maxn],a[maxn];
bool vis[maxn];
struct node
{
	int id;
	ll s;
};
bool operator<(const node &a,const node &b)
{
	return a.s<b.s;
}
priority_queue<node>P;
int main()
{
	int n,k;scanf("%d%d",&n,&k);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
		l[i]=i-1;r[i]=i+1;
		node e;e.id=i;e.s=a[i];
		P.push(e);
	}
	ll ans=0;
	while(k--)
	{
		while(vis[P.top().id]) P.pop();
		node e=P.top();P.pop();
		if(e.s<0)break;
		ans+=e.s;
		e.s=a[e.id]=a[l[e.id]]+a[r[e.id]]-a[e.id];
		P.push(e);
		vis[l[e.id]]=vis[r[e.id]]=1;
		l[e.id]=l[l[e.id]];r[l[e.id]]=e.id;
		r[e.id]=r[r[e.id]];l[r[e.id]]=e.id;
	}
	printf("%lld\n",ans);
	return 0;
}