bzoj 1858: [Scoi2010]序列操作

#include<cstdio>
#define M 100010
inline int max(int a,int b){return a>b?a:b;}
struct tree{int l,r,s,lazy,sum,ml1,mr1,mm1,ml0,mr0,mm0;}tr[M*6];
int a[M],n,m,x,y,v,op;
inline int read()
{
    int tmp=0;
    char ch=getchar();
    while(ch<‘0‘||ch>‘9‘) ch=getchar();
    while(ch>=‘0‘&&ch<=‘9‘){tmp=tmp*10+ch-‘0‘;ch=getchar();}
    return tmp;
}
inline void pu(int p)
{
    int p1=p<<1,p2=p<<1|1;
    tr[p].sum=tr[p1].sum+tr[p2].sum;
    tr[p].mm1=tr[p1].mr1+tr[p2].ml1;
    tr[p].mm0=tr[p1].mr0+tr[p2].ml0;
    if(tr[p1].mr1==tr[p1].s) tr[p].ml1=tr[p1].mr1+tr[p2].ml1;
    else tr[p].ml1=tr[p1].ml1;
    if(tr[p1].mr0==tr[p1].s) tr[p].ml0=tr[p1].mr0+tr[p2].ml0;
    else tr[p].ml0=tr[p1].ml0;
    if(tr[p2].mr1==tr[p2].s) tr[p].mr1=tr[p2].mr1+tr[p1].mr1;
    else tr[p].mr1=tr[p2].mr1;
    if(tr[p2].mr0==tr[p2].s) tr[p].mr0=tr[p2].mr0+tr[p1].mr0;
    else tr[p].mr0=tr[p2].mr0;
    tr[p].mm1=max(tr[p].mm1,max(tr[p1].mm1,tr[p2].mm1));
    tr[p].mm0=max(tr[p].mm0,max(tr[p1].mm0,tr[p2].mm0)); 
}
inline void change(int p)
{
    int t1=tr[p].ml0,t2=tr[p].mr0,t3=tr[p].mm0;
    tr[p].ml0=tr[p].ml1;
    tr[p].ml1=t1;
    tr[p].mr0=tr[p].mr1;
    tr[p].mr1=t2;
    tr[p].mm0=tr[p].mm1;
    tr[p].mm1=t3;
}
inline void make(int l,int r,int p)
{
    tr[p].l=l,tr[p].r=r,tr[p].s=tr[p].r-tr[p].l+1,tr[p].lazy=-1;
    if(l==r)
    {
        tr[p].sum=a[l];
        tr[p].ml1=tr[p].mm1=tr[p].mr1=a[l];
        tr[p].ml0=tr[p].mm0=tr[p].mr0=a[l]^1;
        return;
    }
    int mid=(l+r)>>1;
    make(l,mid,p<<1);
    make(mid+1,r,p<<1|1);
    pu(p);
}
inline void pd(int p)
{
    int lz=tr[p].lazy,p1=p<<1,p2=p<<1|1;
    if(lz==0)
    {
        tr[p1].lazy=tr[p2].lazy=0;
        tr[p1].sum=0;
        tr[p2].sum=0;
        tr[p1].ml1=tr[p1].mr1=tr[p1].mm1=0;
        tr[p2].ml1=tr[p2].mr1=tr[p2].mm1=0;
        tr[p1].ml0=tr[p1].mr0=tr[p1].mm0=tr[p1].s;
        tr[p2].ml0=tr[p2].mr0=tr[p2].mm0=tr[p2].s;
    }
    else if(lz==1)
    {
        tr[p1].lazy=tr[p2].lazy=1;
        tr[p1].sum=tr[p1].s;
        tr[p2].sum=tr[p2].s;
        tr[p1].ml1=tr[p1].mr1=tr[p1].mm1=tr[p1].s;
        tr[p2].ml1=tr[p2].mr1=tr[p2].mm1=tr[p2].s;
        tr[p1].ml0=tr[p1].mr0=tr[p1].mm0=0;
        tr[p2].ml0=tr[p2].mr0=tr[p2].mm0=0;
    }
    else if(lz==2)
    {
        if(tr[p1].lazy==1) tr[p1].lazy=0;
        else if(tr[p1].lazy==0) tr[p1].lazy=1;
        else if(tr[p1].lazy==2) tr[p1].lazy=-1;
        else tr[p1].lazy=2;
        if(tr[p2].lazy==1) tr[p2].lazy=0;
        else if(tr[p2].lazy==0) tr[p2].lazy=1;
        else if(tr[p2].lazy==2) tr[p2].lazy=-1;
        else tr[p2].lazy=2;
        tr[p1].sum=tr[p1].s-tr[p1].sum;
        tr[p2].sum=tr[p2].s-tr[p2].sum;
        change(p1);change(p2);
    }
    tr[p].lazy=-1;
}
inline int find1(int l,int r,int p)
{
    pd(p);
    if(tr[p].l==l&&tr[p].r==r) return tr[p].sum;
    int mid=(tr[p].l+tr[p].r)>>1;
    if(mid>=r) return find1(l,r,p<<1);
    else if(mid<l) return find1(l,r,p<<1|1);
    else return find1(l,mid,p<<1)+find1(mid+1,r,p<<1|1);
}
inline int findl(int l,int r,int p)
{
    pd(p);
    if(tr[p].ml1+l>r) return r-l+1;
    else return tr[p].ml1;
}
inline int findr(int l,int r,int p)
{
    pd(p);
    if(l+tr[p].mr1>r) return r-l+1;
    else return tr[p].mr1;
}
inline int find2(int l,int r,int p)
{
    pd(p);
    if(tr[p].l==l&&tr[p].r==r) return max(tr[p].ml1,max(tr[p].mm1,tr[p].mr1));
    int mid=(tr[p].l+tr[p].r)>>1;
    if(mid>=r) return find2(l,r,p<<1);
    else if(mid<l) return find2(l,r,p<<1|1);
    else return max(max(find2(l,mid,p<<1),find2(mid+1,r,p<<1|1)),findl(mid+1,r,p<<1|1)+findr(l,mid,p<<1));
}
inline void xg(int l,int r,int c,int p)
{
    pd(p);
    if(tr[p].l==l&&tr[p].r==r)
    {
        tr[p].lazy=c;
        if(c==0)
        {
            tr[p].ml1=tr[p].mr1=tr[p].mm1=0;
            tr[p].ml0=tr[p].mr0=tr[p].mm0=tr[p].s;
            tr[p].sum=0;
        }
        else if(c==1)
        {
            tr[p].ml0=tr[p].mr0=tr[p].mm0=0;
            tr[p].ml1=tr[p].mr1=tr[p].mm1=tr[p].s;
            tr[p].sum=tr[p].s;
        }
        else
        {
            tr[p].sum=tr[p].s-tr[p].sum;
            change(p);
        }
        return;
    }
    int mid=(tr[p].l+tr[p].r)>>1;
    if(mid>=r) xg(l,r,c,p<<1);
    else if(mid<l) xg(l,r,c,p<<1|1);
    else 
    {
        xg(l,mid,c,p<<1);
        xg(mid+1,r,c,p<<1|1);
    }
    pu(p);
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++) a[i]=read();
    make(1,n,1);
    for(int i=0;i<m;i++)
    {
        op=read();x=read();y=read();
        if(op<3) xg(x+1,y+1,op,1);
        else if(op==3) printf("%d\n",find1(x+1,y+1,1));
        else printf("%d\n",find2(x+1,y+1,1));
    }
    return 0;
}