【jzoj4726】【NOIP2016提高A组模拟8.22】【种花】【可撤销贪心】

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
#include
#include
#include
#include
#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;}
};
multisets;
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;
}