HDU 3397 Sequence operation

题目意思很简单，四种操作：全部变成1，全部变成0，或者区间异或，查询区间1的个数，和区间连续的1的最大长度。理解起来没有问题，可是写起代码来发现非常繁琐，共用了9个数组，差不多200行了，第一次写这么长得代码，WA一次，后来发现某个地方rt<<1,写成了rt<<1|1,还有PushUp向上更新时忘记了cover[rt]也要根据sum[rt]的值进行更新。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstdlib>
#define N 111111
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;

int sum[N<<2],msum[N<<2],lsum[N<<2],rsum[N<<2],XOR[N<<2],cover[N<<2],msum0[N<<2],lsum0[N<<2],rsum0[N<<2];


//sum表示区间1的个数，msum表示区间连续1的最大长度，lsum表示从左端开始的1的最大长度，
//XOR为异或标记，cover为覆盖标记，1,0表示区间分别被1,0完全覆盖，-1不完全覆盖
//msum0,lsum0,rsum0表示对应的0的个数的统计。


void PushUp(int rt,int m)
{
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
    cover[rt]=((sum[rt]==m)?1:((sum[rt]==0)?0:-1));
    lsum[rt]=lsum[rt<<1];//我擦
    rsum[rt]=rsum[rt<<1|1];
    lsum0[rt]=lsum0[rt<<1];
    rsum0[rt]=rsum0[rt<<1|1];
    if(lsum[rt]==(m-(m>>1)))
        lsum[rt]+=lsum[rt<<1|1];
    else if(lsum0[rt]==(m-(m>>1)))
        lsum0[rt]+=lsum0[rt<<1|1];
    if(rsum[rt]==(m>>1))
        rsum[rt]+=rsum[rt<<1];
    else if(rsum0[rt]==(m>>1))
        rsum0[rt]+=rsum0[rt<<1];
    msum0[rt]=max(max(msum0[rt<<1],msum0[rt<<1|1]),rsum0[rt<<1]+lsum0[rt<<1|1]);
    msum[rt]=max(max(msum[rt<<1],msum[rt<<1|1]),rsum[rt<<1]+lsum[rt<<1|1]);
}

void FXOR(int rt,int m)
{
    if(cover[rt]!=-1)
    {
        cover[rt]^=1;
        sum[rt]=msum[rt]=lsum[rt]=rsum[rt]=m*cover[rt];
        msum0[rt]=lsum0[rt]=rsum0[rt]=m*(1-cover[rt]);
    }
    else
    {
        XOR[rt]^=1;
        sum[rt]=(m-sum[rt]);
        int a=lsum[rt],b=rsum[rt];
        lsum[rt]=lsum0[rt];
        rsum[rt]=rsum0[rt];
        lsum0[rt]=a;
        rsum0[rt]=b;
        a=msum[rt];
        msum[rt]=msum0[rt];
        msum0[rt]=a;
    }
}

void PushDown(int rt,int m)
{
    if(cover[rt]!=-1)
    {
        lsum[rt<<1]=rsum[rt<<1]=msum[rt<<1]=sum[rt<<1]=(m-(m>>1))*cover[rt];
        lsum[rt<<1|1]=rsum[rt<<1|1]=msum[rt<<1|1]=sum[rt<<1|1]=(m>>1)*cover[rt];
        lsum0[rt<<1]=rsum0[rt<<1]=msum0[rt<<1]=(m-(m>>1))*(1-cover[rt]);
        lsum0[rt<<1|1]=rsum0[rt<<1|1]=msum0[rt<<1|1]=(m>>1)*(1-cover[rt]);
        cover[rt<<1]=cover[rt<<1|1]=cover[rt];
        XOR[rt<<1]=XOR[rt<<1|1]=0;
        cover[rt]=-1;
    }
    if(XOR[rt])
    {
        FXOR(rt<<1,(m-(m>>1)));
        FXOR(rt<<1|1,(m>>1));
        XOR[rt]=0;
    }

}

void build(int l,int r,int rt)
{
    XOR[rt]=0;
    if(l==r)
    {
        scanf("%d",&sum[rt]);
        lsum[rt]=rsum[rt]=msum[rt]=cover[rt]=sum[rt];
        lsum0[rt]=rsum0[rt]=msum0[rt]=1-sum[rt];
        return;
    }
    int m=(l+r)>>1;
    build(lson);
    build(rson);
    PushUp(rt,r-l+1);
}

void update(int L,int R,int c,int l,int r,int rt)
{
    if(L<=l&&R>=r)
    {
        if(c==0)
            cover[rt]=0,lsum[rt]=rsum[rt]=sum[rt]=msum[rt]=0,lsum0[rt]=rsum0[rt]=msum0[rt]=r-l+1,XOR[rt]=0;
        else if(c==1)
            cover[rt]=1,lsum[rt]=rsum[rt]=sum[rt]=msum[rt]=r-l+1,lsum0[rt]=rsum0[rt]=msum0[rt]=0,XOR[rt]=0;
        else FXOR(rt,r-l+1);
        return;
    }
    int m=(l+r)>>1;
    PushDown(rt,r-l+1);
    if(L<=m)
        update(L,R,c,lson);
    if(R>m)
        update(L,R,c,rson);
    PushUp(rt,r-l+1);
}

int query1(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
        return sum[rt];
    int m=(l+r)>>1;
    PushDown(rt,r-l+1);
    int res=0;
    if(L<=m)
        res+=query1(L,R,lson);
    if(R>m)
        res+=query1(L,R,rson);
    return res;
}

int query2(int L,int R,int l,int r,int rt)
{
    if(L<=l&&R>=r)
        return msum[rt];
    int m=(l+r)>>1;
    PushDown(rt,r-l+1);
    int ans=0;
    if(R<=m)
        ans=max(ans,query2(L,R,lson));
    else if(L>m)
        ans=max(ans,query2(L,R,rson));
    else
    {
        ans=max(ans,query2(L,R,lson));
        ans=max(ans,query2(L,R,rson));
        int ll,rr;
        ll=min(m-L+1,rsum[rt<<1]);
        rr=min(R-m,lsum[rt<<1|1]);
        ans=max(ans,ll+rr);
    }
    return ans;
}

int main(void)
{
    int tc,n,m;
    int op,a,b,ans;
    scanf("%d",&tc);
    while(tc--)
    {
        scanf("%d%d",&n,&m);
        build(1,n,1);
        while(m--)
        {
            scanf("%d%d%d",&op,&a,&b);
            switch(op)
            {
            case 0:
            case 1:
            case 2:
                update(a+1,b+1,op,1,n,1);
                break;
            case 3:
                ans=query1(a+1,b+1,1,n,1);
                printf("%d
",ans);
                break;
            case 4:
                ans=query2(a+1,b+1,1,n,1);
                printf("%d
",ans);
                break;
            }
        }
    }
    return 0;
}