标签:sample ack lib mat .com script wim 负数 color
Description
在长度为的序列中选出段互不相交的子段,求最大字段和.
Input
第一行两个整数.
第二行个整数.
Output
一行一个整数表示最大值.
Sample Input
10 -1 10 -1 10
Sample Output
29
HINT
Solution
如果序列中正整数个数,直接取最大的个数的和即可.
将序列合并成若干个交错的正负段和,如-1,-2,3,4,-5,-6可以合并成-1-2,3+4,-5-6.
记录所有正数段之和,设正数段个数为,则需要把k段正数段合并成段.
对于每一段正数段有两种操作:
1.舍弃;
2.与相邻串合并.
可以用堆来实现:将所有段的绝对值扔入堆中,每次取最小的元素(如果在当前序列中是左或右没有元素的负数段就跳过),,将与在当前序列中的左右元素合并即可.
#include<cmath> #include<ctime> #include<stack> #include<queue> #include<cstdio> #include<vector> #include<cstring> #include<cstdlib> #include<iostream> #include<algorithm> #define N 100005 #define M -100000000000001LL using namespace std; typedef long long ll; struct heap{ ll k;int n; }h[N],tmp; ll a[N],num[N],sum,ans; int lst[N],nxt[N],n,m,cnt; inline bool cmp(ll x,ll y){ return x>y; } inline bool chk(heap x,heap y){ if(x.k!=y.k) return x.k<y.k; return x.n<y.n; } inline void swim(int i){ heap a=h[i];int j=i>>1; while(j&&chk(a,h[j])){ h[i]=h[j];num[h[i].n]=i;i=j;j>>=1; } h[i]=a;num[h[i].n]=i; } inline void sink(int i){ heap a=h[i];int j=i<<1; while(j<=cnt){ if(j<cnt&&chk(h[j+1],h[j])) ++j; if(chk(a,h[j])) break; h[i]=h[j];num[h[i].n]=i;i=j;j<<=1; } h[i]=a;num[h[i].n]=i; } inline void del(int u){ if(!u||u>n) return; if(num[u]){ if(num[u]==cnt) --cnt; else{ h[num[u]]=h[cnt--];sink(num[u]);swim(num[u]); } } lst[nxt[u]]=lst[u];nxt[lst[u]]=nxt[u]; } inline void init(){ scanf("%d%d",&n,&m); for(int i=1;i<=n;++i){ scanf("%lld",&a[i]); if(a[i]>=0LL){ ++cnt;ans+=a[i]; } } /*正数个数<=m*/ if(cnt<=m){ sort(a+1,a+1+n,cmp); for(int i=1;i<=m;++i) sum+=a[i]; printf("%lld\n",sum); return; } cnt=0; for(int i=1;i<=n;++i){ if(sum*a[i]<0LL){ a[++cnt]=sum;sum=a[i]; } else sum+=a[i]; } a[++cnt]=sum; n=cnt;cnt=0; for(int i=1;i<=n;++i) if(a[i]>0) ++cnt; /*正数区间数<=m*/ if(cnt<=m){ printf("%lld\n",ans);return; } for(int i=1;i<=n;++i){ lst[i]=i-1;nxt[i]=i+1; } lst[n+1]=n;nxt[0]=1; m=cnt-m;cnt=0; for(int i=1;i<=n;++i){ h[++cnt].k=abs(a[i]);h[cnt].n=i;num[i]=cnt;swim(cnt); } while(m){ tmp=h[1];h[1]=h[cnt--];sink(1); if(((!lst[tmp.n]||nxt[tmp.n]>n)&&a[tmp.n]<0)||!a[tmp.n]) continue; ans-=tmp.k;a[tmp.n]+=a[lst[tmp.n]]+a[nxt[tmp.n]]; del(lst[tmp.n]);del(nxt[tmp.n]); h[++cnt].k=abs(a[tmp.n]);h[cnt].n=tmp.n;swim(cnt); --m; } printf("%lld\n",ans); } int main(){ freopen("sequence.in","r",stdin); freopen("sequence.out","w",stdout); init(); fclose(stdin); fclose(stdout); return 0; }
标签:sample ack lib mat .com script wim 负数 color
原文地址:http://www.cnblogs.com/AireenYe/p/6075823.html