题目描述
A城市有一个巨大的圆形广场,为了绿化环境和净化空气,市政府决定沿圆形广场外圈种一圈树。
园林部门得到指令后,初步规划出n个种树的位置,顺时针编号1到n。并且每个位置都有一个美观度Ai,如果在这里种树就可以得到这Ai的美观度。但由于A城市土壤肥力欠佳,两棵树决不能种在相邻的位置(i号位置和i+1号位置叫相邻位置。值得注意的是1号和n号也算相邻位置!)。
最终市政府给园林部门提供了m棵树苗并要求全部种上,请你帮忙设计种树方案使得美观度总和最大。如果无法将m棵树苗全部种上,给出无解信息。
输入输出格式
输入格式:
输入的第一行包含两个正整数n、m。第二行n个整数Ai。
输出格式:
输出一个整数,表示最佳植树方案可以得到的美观度。如果无解输出“Error!”,不包含引号。
输入输出样例
输入样例#1:
7 3
1 2 3 4 5 6 7
输出样例#1:
15
输入样例#2:
7 4
1 2 3 4 5 6 7
输出样例#2:
Error!
说明
对于全部数据:m<=n;
-1000<=Ai<=1000
刚看到这题第一反应是DP。。然而瞅了一眼数据范围(emm,200000)
发现是一个贪心。。思路非常神奇。。
就是以在每个坑种树的收益建一个大根堆
然后用链表存它的前驱后继(n的后继为1,1的前驱为n)
每次取最大的收益。
那么问题来了————有可能在这个点种树比在它两边各种一棵树收益小。
但一定是两边各种一棵,因为我们建的是大根堆,先出来的一定大。
所以我们可以选这个坑种树,也可以选两边的坑各种一棵,所以我们给我们的贪心一个反悔的机会
就是选在这个坑种树以后将它两边的坑合并为一个价值为(w[l]+w[r]-w[now])的坑,代替我们选择的坑
这里非常的巧妙,就是如果选了这个坑就相当于不在那个中间的坑种树了,而是在它两边各种一棵。
#include<cstdio>
#include<cstring>
#include<cmath>
#include<iostream>
#include<algorithm>
#include<queue>
# define M (200000+50)
using namespace std;
struct node{
int value,place;
friend bool operator < (node x,node y){
return x.value < y.value;
}
};
priority_queue<node> q;
int n,m,Ans,chose,nxt[M],pre[M],w[M],l,r;
bool vis[M];
void change(int x){
vis[x]=1;
nxt[pre[x]]=nxt[x];
pre[nxt[x]]=pre[x];
nxt[x]=0; pre[x]=0;
}
int main(){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {
scanf("%d",&w[i]);
}
if((n>>1)<m){
cout<<"Error!"<<endl;
return 0;
}
for(int i=1;i<n;i++) nxt[i]=i+1; nxt[n]=1;
for(int i=2;i<=n;i++) pre[i]=i-1; pre[1]=n;
for(int i=1;i<=n;i++) q.push((node){w[i],i});
for(int i=1;i<=m;i++){
while(vis[q.top().place]) q.pop() ;
node temp = q.top(); q.pop();
Ans+= temp.value;
l=pre[temp.place]; r=nxt[temp.place];
w[temp.place]=w[l]+w[r]-w[temp.place];
change(l); change(r);
q.push((node){w[temp.place],temp.place});
}
printf("%d\n",Ans);
return 0;
}