description
经过三十多个小时的长途跋涉,小Z和小D终于到了NOI现场——南山南中学。一进校园,小D就被花所吸引了(不要问我为什么),遍和一旁的种花园丁交(J)流(L)了起来。
他发现花的摆放竟有如此奥秘:圆形广场共有 N 个种花的位置,顺时针编号1到N。并且每个位置都有一个美观度ai ,如果在这里种花就可以得到这ai 的美观度。但由于地处南山土壤肥力欠佳,两株花不能种在相邻的位置(1号和N号也算相邻位置)。校方一共给了 M 株花,经过园丁的精妙摆放,才能如此吸引小D。所以现在小D也想知道应该如何摆这 N 株花。
solution
可以发现先贪心地取,a[i]表示改变当前状态的贡献,间隔地取,之前取了的现在不取,之前不取的现在取,保证左右边界取。这样贪心去了当前a[i],再在堆里加上a[left]+a[right]-a[i]。注意下一次就不能选left,right,要打一个标记去除掉。
code
#include<set>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>
#define LF double
#define LL long long
#define ULL unsigned LL
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
#define fr(i,j) for(int i=begin[j];i;i=next[i])
using namespace std ;
int const mn=2 *1e5 +3 ,inf=1e9 +7 ;
int n,m,a[mn],pre[mn],next[mn],tag[mn];
struct rec{
int p;
friend bool operator <(rec x,rec y){
return a[x.p]>a[y.p];
}
rec(int P){p=P;}
};
multiset <rec> s;
void dele(int x){
int pr=pre[x],ne=next[x];
next[pr]=ne;
pre[ne]=pr;
}
int main(){
freopen("d.in" ,"r" ,stdin);
freopen("d.out" ,"w" ,stdout);
scanf ("%d%d" ,&n,&m);
fo(i,1 ,n){
scanf ("%d" ,&a[i]);
pre[i]=(i-1 )?i-1 :n;
next[i]=(i!=n)?i+1 :1 ;
s.insert(rec(i));
}
int ans=0 ;
if (m>n/2 ){
printf ("Error!" );
return 0 ;
}
fo(i,1 ,m){
rec tmp=*s.begin();
while (tag[tmp.p]){
s.erase(s.begin());
tmp=*s.begin();
}
s.erase(s.begin());
ans+=a[tmp.p];
a[tmp.p]=a[pre[tmp.p]]+a[next[tmp.p]]-a[tmp.p];
s.insert(rec(tmp.p));
tag[pre[tmp.p]]=tag[next[tmp.p]]=1 ;
dele(pre[tmp.p]);
dele(next[tmp.p]);
}
printf ("%d" ,ans);
return 0 ;
}