【贪心+优先队列】51Nod 1053 最大M子段和 V2

题面在这里

其实很简单……

首先有一个贪心想法就是把所有的正数都取来

但是这样子段数可能会超过m

那么有以下两种方法减少子段数：

删去一个正的子段

取一个负的子段，并将相邻两个正子段合并

然后把这两种操作都扔到优先队列里，每次取就好了

示例程序：

#include<cstdio>
#include<algorithm>
#include<queue>
#define _abs(x) ((x)>0?(x):-(x))
using namespace std;
typedef long long ll;
inline char nc(){
static char buf[100000],*p1=buf,*p2=buf;
return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int red(){
int res=0,f=1;char ch=nc();
while (ch<'0'||'9'<ch) {if (ch=='-') f=-f;ch=nc();}
while ('0'<=ch&&ch<='9') res=(res<<3)+(res<<1)+ch-48,ch=nc();
return res*f;
}

const int maxn=50005;
int n,N,m,a[maxn],lst[maxn],nxt[maxn],tot;
ll ans,b[maxn];
inline void del(int i){
lst[nxt[i]]=lst[i];
nxt[lst[i]]=nxt[i];
}
struct data{
ll x;int id;
data () {}
data(ll _x,int _id):x(_x),id(_id) {}
bool operator<(const data&b)const{
return x>b.x;
}
};
priority_queue<data> Q;
int main(){
n=red(),m=red();
for (int i=1;i<=n;i++){
a[i]=red();
if (i==1||(a[i]>0)^(a[i-1]>0)) N++;
b
+=a[i];
}
for (int i=1;i<=N;i++){
lst[i]=i-1;nxt[i]=i+1;
Q.push(data(_abs(b[i]),i));
if (b[i]>0) ans+=b[i],tot++;
}
lst[0]=0;nxt[0]=1;lst[N+1]=N;nxt[N+1]=N+1;
while (tot>m){
data d=Q.top();Q.pop();
if (lst[nxt[d.id]]!=d.id||nxt[lst[d.id]]!=d.id) continue;
if (b[d.id]<=0&&(lst[d.id]<1||nxt[d.id]>N)) continue;
ans-=d.x;tot--;b[d.id]+=b[lst[d.id]]+b[nxt[d.id]];del(lst[d.id]);del(nxt[d.id]);
Q.push(data(_abs(b[d.id]),d.id));
}
printf("%lld",ans);
return 0;
}