bzoj2288【POJ Challenge】生日礼物（贪心+堆）

首先我们把连续的正数/负数合成一个，因为显然要取都取。
然后开头结尾的负数可以去掉了，我们一定不会取。
这样我们就得到了正负正负…正的一个序列。记正数有cnt个，则若cnt<=m，把正数全取了即可。否则，m=cnt-m,我们需要做出一些牺牲。可以删去一个正数或者选上一个负数来使得两边的正数合成一个，共需要m次牺牲，我们肯定是要贪心取牺牲最小的。还有一个限制就是如果我们选了一个正数，则他两边的负数都不能选；如果我们选了一个负数，则他两边的正数都不能选。也就是相邻两数不能都选。这样就转化成了类似bzoj1150的问题。可以贪心+堆来解决。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define inf 0x3f3f3f3f
#define N 100010
inline char gc(){
    static char buf[1<<16],*S,*T;
    if(S==T){T=(S=buf)+fread(buf,1,1<<16,stdin);if(T==S) return EOF;}
    return *S++;
}
inline int read(){
    int x=0,f=1;char ch=gc();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=gc();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=gc();
    return x*f;
}
int n,m,a[N],ans=0,w[N],tot=0,cnt=0,pre[N],nxt[N];
bool del[N];
struct cmp{
    bool operator()(int x,int y){return w[x]>w[y];}
};priority_queue<int,vector<int>,cmp>q;
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<=n;++i) a[i]=read();
    int l=1;while(l<=n&&a[l]<0) ++l;int op=1,sum=0;
    for(int i=l;i<=n;++i){
        if(op*a[i]<0){
            w[++tot]=sum*op;q.push(tot);
            if(op==1) ++cnt,ans+=sum;sum=0;op=-op;
        }sum+=a[i];
    }if(op==1){w[++tot]=sum;q.push(tot);++cnt;ans+=sum;}
    if(cnt<=m){printf("%d\n",ans);return 0;}w[0]=inf;
    m=cnt-m;for(int i=1;i<=tot;++i) pre[i]=i-1,nxt[i]=i+1>tot?0:i+1;
    while(m--){
        int x=q.top();q.pop();while(del[x]){x=q.top();q.pop();}
        ans-=w[x];w[x]=w[pre[x]]+w[nxt[x]]-w[x];q.push(x);
        del[pre[x]]=1;del[nxt[x]]=1;pre[x]=pre[pre[x]];nxt[x]=nxt[nxt[x]];
        nxt[pre[x]]=x;pre[nxt[x]]=x;
    }printf("%d\n",ans);
    return 0;
}