<JZOJ4726>种花

挺有意思的贪心

神奇的贪心

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<queue>
#define rint register int
using std::priority_queue;
template <class T>inline void read(T &X)
{
    X=;int W=;char ch=;
    while(!isdigit(ch))W|=ch=='-',ch=getchar();
    while(isdigit(ch))X=(X<<)+(X<<)+(ch^),ch=getchar();
    X=W?-X:X;return;
}
int n,m,pre[],nxt[],vis[]={};
long long ans=,a[];
struct node
{
    int num;
    friend bool operator <(node x,node y)
    {
        return a[x.num]<a[y.num];
    }
};
priority_queue<node>q;
int main()
{
    read(n),read(m);
    if(m>(n>>)){printf("Error!");return ;}
    node x;
    for(rint i=;i<=n;++i)
    {
        read(a[i]);
        pre[i]=(i==)?n:i-,nxt[i]=(i==n)?:i+;
        x.num=i,q.push(x);
    } 

    while(m--)
    {
        for(x=q.top(),q.pop();vis[x.num];x=q.top(),q.pop());
        ans+=a[x.num];
        a[x.num]=a[pre[x.num]]+a[nxt[x.num]]-a[x.num];
        vis[pre[x.num]]=vis[nxt[x.num]]=;
        pre[x.num]=pre[pre[x.num]];
        nxt[x.num]=nxt[nxt[x.num]];
        nxt[pre[x.num]]=pre[nxt[x.num]]=x.num;
        q.push(x);
    }
    printf("%lld\n",ans);
return ;
}