HDU 3397 Sequence operation

很麻烦的一道线段树。。。

#include <algorithm>
#include <cstring>
#include <cstdio>
#include <vector>
#include <queue>
#include <cmath>
using namespace std;
struct SEG
{
    int l, r, va;
    int le, re, ml, sum1;
    int lz, rz, mz, sum2;
    int rev;
} seg[100005<<3];
void update(int rt)
{
    int le, re, ml, ll, rr;
    if(seg[rt].l==seg[rt].r) return;
    le=seg[rt<<1].le, re=seg[rt<<1|1].re;
    ml=max(seg[rt<<1].ml,seg[rt<<1|1].ml);
    ml=max(ml,seg[rt<<1].re+seg[rt<<1|1].le);
    seg[rt].le=seg[rt<<1].le;
    seg[rt].re=seg[rt<<1|1].re;
    ll=seg[rt<<1].r-seg[rt<<1].l+1, rr=seg[rt<<1|1].r-seg[rt<<1|1].l+1;
    if(le==ll)
        seg[rt].le+=seg[rt<<1|1].le;
    if(re==rr)
        seg[rt].re+=seg[rt<<1].re;
    seg[rt].ml=ml;
    //处理0
    int lz, rz, mz;
    lz=seg[rt<<1].lz, rz=seg[rt<<1|1].rz;
    mz=max(seg[rt<<1].mz,seg[rt<<1|1].mz);
    mz=max(mz,seg[rt<<1].rz+seg[rt<<1|1].lz);
    seg[rt].lz=seg[rt<<1].lz;
    seg[rt].rz=seg[rt<<1|1].rz;
    ll=seg[rt<<1].r-seg[rt<<1].l+1, rr=seg[rt<<1|1].r-seg[rt<<1|1].l+1;
    if(lz==ll)
        seg[rt].lz+=seg[rt<<1|1].lz;
    if(rz==rr)
        seg[rt].rz+=seg[rt<<1].rz;
    seg[rt].mz=mz;

    seg[rt].sum1=seg[rt<<1].sum1+seg[rt<<1|1].sum1;
    seg[rt].sum2=seg[rt].r-seg[rt].l+1-seg[rt].sum1;
}
void push_down(int rt)
{
    int tl, tr, m;
    tl=seg[rt].l, tr=seg[rt].r;
    m=(tl+tr)>>1;
    if(tl==tr) return;
    if(seg[rt].va!=-1)
    {
        seg[rt<<1].va=seg[rt<<1|1].va=seg[rt].va;
        seg[rt<<1].rev=seg[rt<<1|1].rev=0;
        if(seg[rt].va==1)
        {
            seg[rt<<1].sum1=seg[rt<<1].le=seg[rt<<1].re=seg[rt<<1].ml=m-tl+1;
            seg[rt<<1|1].sum1=seg[rt<<1|1].le=seg[rt<<1|1].re=seg[rt<<1|1].ml=tr-m;
            seg[rt<<1].sum2=seg[rt<<1].lz=seg[rt<<1].rz=seg[rt<<1].mz=0;
            seg[rt<<1|1].sum2=seg[rt<<1|1].lz=seg[rt<<1|1].rz=seg[rt<<1|1].mz=0;
        }
        else if(seg[rt].va==0)
        {
            seg[rt<<1].sum1=seg[rt<<1].le=seg[rt<<1].re=seg[rt<<1].ml=0;
            seg[rt<<1|1].sum1=seg[rt<<1|1].le=seg[rt<<1|1].re=seg[rt<<1|1].ml=0;
            seg[rt<<1].sum2=seg[rt<<1].lz=seg[rt<<1].rz=seg[rt<<1].mz=m-tl+1;
            seg[rt<<1|1].sum2=seg[rt<<1|1].lz=seg[rt<<1|1].rz=seg[rt<<1|1].mz=tr-m;
        }
        seg[rt].va=-1;
    }
    if(seg[rt].rev)
    {
        seg[rt].rev=0;
        seg[rt<<1].rev^=1, seg[rt<<1|1].rev^=1;
        swap(seg[rt<<1].le,seg[rt<<1].lz);
        swap(seg[rt<<1].re,seg[rt<<1].rz);
        swap(seg[rt<<1].ml,seg[rt<<1].mz);
        swap(seg[rt<<1].sum1,seg[rt<<1].sum2);
        swap(seg[rt<<1|1].le,seg[rt<<1|1].lz);
        swap(seg[rt<<1|1].re,seg[rt<<1|1].rz);
        swap(seg[rt<<1|1].ml,seg[rt<<1|1].mz);
        swap(seg[rt<<1|1].sum1,seg[rt<<1|1].sum2);
    }
}
void build(int l,int r,int rt)
{
    seg[rt].l=l, seg[rt].r=r, seg[rt].va=-1, seg[rt].rev=0;
    if(l==r)
    {
        scanf("%d",&seg[rt].va);
        seg[rt].sum1=seg[rt].va;
        seg[rt].sum2=1-seg[rt].va;
        seg[rt].ml=seg[rt].re=seg[rt].le=seg[rt].va;
        seg[rt].mz=seg[rt].rz=seg[rt].lz=1-seg[rt].va;
        return;
    }
    int m=(l+r)>>1;
    build(l,m,rt<<1);
    build(m+1,r,rt<<1|1);
    update(rt);
}
void change(int l,int r,int va,int rt)
{
    push_down(rt);
    int tl=seg[rt].l, tr=seg[rt].r;
    int m=(tl+tr)>>1;
    if(l<=tl&&tr<=r)
    {
        if(va==1)
        {
            seg[rt].va=va;
            seg[rt].sum1=seg[rt].le=seg[rt].re=seg[rt].ml=tr-tl+1;
            seg[rt].sum2=seg[rt].lz=seg[rt].rz=seg[rt].mz=0;
        }
        else if(va==0)
        {
            seg[rt].va=va;
            seg[rt].sum1=seg[rt].le=seg[rt].re=seg[rt].ml=0;
            seg[rt].sum2=seg[rt].lz=seg[rt].rz=seg[rt].mz=tr-tl+1;
        }
        else
        {
            seg[rt].rev=1;
            swap(seg[rt].le,seg[rt].lz);
            swap(seg[rt].re,seg[rt].rz);
            swap(seg[rt].ml,seg[rt].mz);
            swap(seg[rt].sum1,seg[rt].sum2);
        }
        return;
    }
    if(l<=m)
        change(l,r,va,rt<<1);
    if(r>m)
        change(l,r,va,rt<<1|1);
    update(rt);
}
int query(int l,int r,int rt)
{
    push_down(rt);
    int tl=seg[rt].l, tr=seg[rt].r;
    if(l<=tl&&tr<=r)
        return seg[rt].ml;
    int m=(tl+tr)>>1;
    int ans;
    if(r<=m)
        ans=query(l,r,rt<<1);
    else if(l>m)
        ans=query(l,r,rt<<1|1);
    else
        ans=max(min(m-l+1,seg[rt<<1].re)+min(r-m,seg[rt<<1|1].le),max(query(l,r,rt<<1),query(l,r,rt<<1|1)));
    return ans;
}
int query2(int l,int r,int rt)
{
    push_down(rt);
    int tl=seg[rt].l, tr=seg[rt].r;
    if(l<=tl&&tr<=r)
        return seg[rt].sum1;
    int m=(tl+tr)>>1;
    int ans=0;
    if(l<=m)
        ans+=query2(l,r,rt<<1);
    if(r>m)
        ans+=query2(l,r,rt<<1|1);
    return ans;
}
int main()
{
    int i, j, n, m, op, a, b, T;
    while(~scanf("%d",&T))
    {
        while(T--)
        {
            scanf("%d%d",&n,&m);
            build(1,n,1);
            while(m--)
            {
                scanf("%d%d%d",&op,&a,&b);
                if(op==0)
                    change(a+1,b+1,0,1);
                else if(op==1)
                    change(a+1,b+1,1,1);
                else if(op==2)
                    change(a+1,b+1,2,1);
                else if(op==3)
                    printf("%d\n",query2(a+1,b+1,1));
                else
                    printf("%d\n",query(a+1,b+1,1));
            }
        }
    }
    return 0;
}