洛谷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

说明

对于20%的数据，n<=20。

对于50%的数据，n<=6000。

对于100%的数据，n<=500000，k<=n/2,在一个地方种树获利的绝对值在1000000以内。

堆，即优先队列。

如果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]，重新找最大的。

记得开 long long !

附代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<queue>
#define MAXN 500010
using namespace std;
int n,m,l[MAXN],r[MAXN];
long long ans=0,a[MAXN];
bool used[MAXN];
struct node{
	long long num;
	int id;
	bool operator <(const node &y)const{
		return num<y.num;
	}
}x;
priority_queue<node> q;
inline int read(){
	int date=0,w=1;char c=0;
	while(c<'0'||c>'9'){if(c=='-')w=-1;c=getchar();}
	while(c>='0'&&c<='9'){date=date*10+c-'0';c=getchar();}
	return date*w;
}
int main(){
	n=read();m=read();
	r[0]=1;l[n+1]=n;
	for(int i=1;i<=n;i++){
		x.num=a[i]=read();
		x.id=i;
		l[i]=i-1;r[i]=i+1;
		used[i]=false;
		q.push(x);
	}
	while(m--){
		for(node i=q.top();used[i.id];i=q.top())q.pop();
		x=q.top();
		q.pop();
		if(x.num<=0)break;
		ans+=x.num;
		a[x.id]=a[l[x.id]]+a[r[x.id]]-a[x.id];
		x.num=a[x.id];
		used[l[x.id]]=used[r[x.id]]=true;
		l[x.id]=l[l[x.id]];r[l[x.id]]=x.id;
		r[x.id]=r[r[x.id]];l[r[x.id]]=x.id;
		q.push(x);
	}
	printf("%lld\n",ans);
	return 0;
}