【CF280D】k-Maximum Subsequence Sum（大码量多细节线段树）

点此看题面

大致题意： 给你一个序列，让你支持单点修改以及询问给定区间内选出至多\(k\)个不相交子区间和的最大值。

题意转换

这道题看似很不可做，实际上可以通过一个简单转换让其变可做。

考虑每次选出一个区间，统计答案后将这个区间内所有数乘上\(-1\)，再继续求答案，这样做是显然合法的。

于是就变成了在给定区间内选出权值和最大子区间和区间乘\(-1\)两种操作。

具体实现

具体实现十分复杂。。。

对于线段树的每个节点，我们需要维护区间和、最大值及其区间、前缀最大值及其区间、后缀最大值及其区间。

但由于要乘\(-1\)，因此我们同时还要维护最小值及其区间、前缀最小值及其区间、后缀最小值及其区间，这样乘\(-1\)后只要交换最大值和最小值即可方便地维护信息。

口胡起来很简洁，但这么多信息实现起来就需要点技巧了。

对于我写的class套struct套struct套struct实在不想多说。。。

代码

#include<bits/stdc++.h>
#define Tp template<typename Ty>
#define Ts template<typename Ty,typename... Ar>
#define Reg register
#define RI Reg int
#define Con const
#define CI Con int&
#define I inline
#define W while
#define N 100000
#define K 20
#define max(x,y) ((x)<(y)?(y):(x))
#define Gmax(x,y) (x<(y)&&(x=(y),0))
using namespace std;
int n,a[N+5];struct Op {int l,r;I Op(CI x=0,CI y=0):l(x),r(y){}}s[K+5];
class FastIO
{
    private:
        #define FS 100000
        #define tc() (A==B&&(B=(A=FI)+fread(FI,1,FS,stdin),A==B)?EOF:*A++)
        #define pc(c) (C==E&&(clear(),0),*C++=c)
        #define tn (x<<3)+(x<<1)
        #define D isdigit(c=tc())
        int f,T;char c,*A,*B,*C,*E,FI[FS],FO[FS],S[FS];
    public:
        I FastIO() {A=B=FI,C=FO,E=FO+FS;}
        Tp I void read(Ty& x) {x=0,f=1;W(!D) f=c^'-'?1:-1;W(x=tn+(c&15),D);x*=f;}
        Ts I void read(Ty& x,Ar&... y) {read(x),read(y...);}
        Tp I void write(Ty x) {W(S[++T]=x%10+48,x/=10);W(T) pc(S[T--]);}
        Tp I void writeln(Con Ty& x) {write(x),pc('\n');}
        I void clear() {fwrite(FO,1,C-FO,stdout),C=FO;}
}F;
struct data//记录某个权值及其区间
{
    int l,r,v;I data(CI x=0,CI y=0,CI z=0):l(x),r(y),v(z){}
    I data operator + (Con data& o) Con {return data(l,o.r,v+o.v);}
    I bool operator < (Con data& o) Con {return v<o.v;}
    I void operator *= (CI o) {v*=o;}
};
struct status//记录一个区间最值、前缀最值、后缀最值
{
    data Pre,Nxt,Val;I status() {Pre=Nxt=Val=data();}
    I void operator *= (CI o) {Pre*=o,Nxt*=o,Val*=o;}
    I data operator = (Con data& o) {return Pre=Nxt=Val=o;}
};
struct qdata//由于询问时需要用到区间和，所以将一个status（最值信息）和一个data（和）绑一起
{
    status Max;data Sum;
    I qdata(Con status& x=status(),Con data& y=data()):Max(x),Sum(y){}
};
class SegmentTree//线段树
{
    private:
        #define P CI l=1,CI r=n,CI rt=1
        #define L l,mid,rt<<1
        #define R mid+1,r,rt<<1|1
        #define LS(x) O[x<<1]
        #define RS(x) O[x<<1|1]
        #define PU(x)\
        (\
            O[x].Max.Pre=max(LS(x).Max.Pre,LS(x).Sum+RS(x).Max.Pre),\
            O[x].Max.Nxt=max(RS(x).Max.Nxt,LS(x).Max.Nxt+RS(x).Sum),\
            O[x].Max.Val=max(LS(x).Max.Nxt+RS(x).Max.Pre,max(LS(x).Max.Val,RS(x).Max.Val)),\
            O[x].Min.Pre=min(LS(x).Min.Pre,LS(x).Sum+RS(x).Min.Pre),\
            O[x].Min.Nxt=min(RS(x).Min.Nxt,LS(x).Min.Nxt+RS(x).Sum),\
            O[x].Min.Val=min(LS(x).Min.Nxt+RS(x).Min.Pre,min(LS(x).Min.Val,RS(x).Min.Val)),\
            O[x].Sum=LS(x).Sum+RS(x).Sum\
        )//上传信息，应该可以再套用define更简短地实现，但我不会。。。
        #define PD(x) !~O[x].F&&(Upt(x<<1),Upt(x<<1|1),O[x].F=1)//下传乘-1的标记
        #define Assign(x,v) (O[x].Sum=O[x].Max=O[x].Min=data(l,r,v))//赋值
        #define Upt(x) (O[x].Sum*=-1,O[x].Max*=-1,O[x].Min*=-1,swap(O[x].Max,O[x].Min),O[x].F*=-1)//乘-1，交换最值
        struct Il {data Sum;status Max,Min;int F;I Il() {Sum=data(),Max=Min=status(),F=1;}}O[N<<2];//维护线段树上节点信息
    public:
        I void Build(P) {if(l==r) return (void)Assign(rt,a[l]);RI mid=l+r>>1;Build(L),Build(R),PU(rt);}//初始化建树
        I void Update(CI x,CI v,P)//单点修改
        {
            if(l==r) return (void)Assign(rt,v);PD(rt);RI mid=l+r>>1;
            x<=mid?Update(x,v,L):Update(x,v,R),PU(rt);
        }
        I void Modify(CI tl,CI tr,P)//区间乘-1
        {
            if(tl<=l&&r<=tr) return (void)Upt(rt);PD(rt);RI mid=l+r>>1;
            tl<=mid&&(Modify(tl,tr,L),0),tr>mid&&(Modify(tl,tr,R),0),PU(rt);
        }
        I qdata Query(CI tl,CI tr,P)//询问
        {
            if(tl<=l&&r<=tr) return qdata(O[rt].Max,O[rt].Sum);PD(rt);RI mid=l+r>>1;
            if(tr<=mid) return Query(tl,tr,L);if(tl>mid) return Query(tl,tr,R);
            qdata ql=Query(tl,tr,L),qr=Query(tl,tr,R),res;//合并左、右子树答案
            res.Max.Pre=max(ql.Max.Pre,ql.Sum+qr.Max.Pre),
            res.Max.Nxt=max(qr.Max.Nxt,ql.Max.Nxt+qr.Sum),
            res.Max.Val=max(ql.Max.Nxt+qr.Max.Pre,max(ql.Max.Val,qr.Max.Val));
            return res.Sum=ql.Sum+qr.Sum,res;
        }
}S;
int main()
{
    RI Qtot,i,op,x,y,z,ans,cnt;data t;for(F.read(n),i=1;i<=n;++i) F.read(a[i]);
    S.Build(),F.read(Qtot);W(Qtot--)
    {
        if(F.read(op,x,y),op)//处理询问
        {
            F.read(z),ans=cnt=0;W(z--) 
            {
                if((t=S.Query(x,y).Max.Val).v>0) S.Modify(t.l,t.r),s[++cnt]=Op(t.l,t.r),ans+=t.v;//求答案，乘-1
                else break;//如果小于0就退出
            }
            W(cnt) S.Modify(s[cnt].l,s[cnt].r),--cnt;F.writeln(ans);//撤销乘-1，输出答案
        }else S.Update(x,y);//单点修改
    }return F.clear(),0;
}

转载于:https://www.cnblogs.com/chenxiaoran666/p/CF280D.html