bzoj1858: [Scoi2010]序列操作

bzoj1858: [Scoi2010]序列操作
一道裸线段树。的确是比较好想，然而代码写得莫名丑，于是调了很长时间。

---

题解

维护区间中1的总数，左起连续1的个数，右起连续1的个数，最大连续1的个数，0同理。更新的时候左起连续1（0）要考虑左区间全为1（0）延伸到右区间的情况，右起同理。最大连续考虑左区间的右连续+右区间的左连续情况。标记打区间填充和取反。

---

代码

#include<iostream>
#include<cstdio>
#define mxn 100010
using namespace std;
int op,x,y,n,m;
bool a[mxn];
struct node
{int ll,rr,sum[2],llen[2],rlen[2],mxlen[2],laz;}b[mxn<<2];
void update(int k)
{
    for(int i=0;i<2;i++)
    {
        b[k].sum[i]=b[k<<1].sum[i]+b[k<<1|1].sum[i];
        b[k].llen[i]=b[k<<1].llen[i],b[k].rlen[i]=b[k<<1|1].rlen[i];
        if(b[k<<1].llen[i]==b[k<<1].rr-b[k<<1].ll+1)
        b[k].llen[i]=b[k<<1].rr-b[k<<1].ll+1+b[k<<1|1].llen[i];
        if(b[k<<1|1].rlen[i]==b[k<<1|1].rr-b[k<<1|1].ll+1)
        b[k].rlen[i]=b[k<<1|1].rr-b[k<<1|1].ll+1+b[k<<1].rlen[i];
        b[k].mxlen[i]=b[k<<1].rlen[i]+b[k<<1|1].llen[i];
        if(b[k].mxlen[i]<b[k<<1].mxlen[i])b[k].mxlen[i]=b[k<<1].mxlen[i];
        if(b[k].mxlen[i]<b[k<<1|1].mxlen[i])b[k].mxlen[i]=b[k<<1|1].mxlen[i];
    }
    b[k].laz=-1;
}
void pushdown(int k);
void build(int k,int l,int r)
{
    b[k].ll=l,b[k].rr=r,b[k].laz=-1;
    if(l==r)
    for(int i=0;i<2;i++)
    b[k].sum[i]=b[k].rlen[i]=b[k].mxlen[i]=b[k].llen[i]=!(a[l]^i);
    else
    {
        int mid=(l+r)>>1;
        build(k<<1,l,mid);
        build(k<<1|1,mid+1,r);
        update(k);
    }
}
void mdf(int k,int l,int r,bool v)
{
    if(b[k].ll==l&&b[k].rr==r)
    {
        b[k].laz=v;
        for(int i=0;i<2;i++)
        b[k].sum[i]=b[k].llen[i]=b[k].rlen[i]=b[k].mxlen[i]=(b[k].rr-b[k].ll+1)*!(v^i);
    }
    else
    {
        pushdown(k);
        int mid=(b[k].ll+b[k].rr)>>1;
        if(l>mid)mdf(k<<1|1,l,r,v);
        else if(r<=mid)mdf(k<<1,l,r,v);
        else mdf(k<<1,l,mid,v),mdf(k<<1|1,mid+1,r,v);
        update(k);
    }
}
void rev(int k,int l,int r)
{
    if(b[k].ll==l&&b[k].rr==r)
    {
        if(b[k].laz==2)b[k].laz=-1;
        else if(b[k].laz>=0)b[k].laz^=1;
        else b[k].laz=2;
        b[k].sum[0]^=b[k].sum[1]^=b[k].sum[0]^=b[k].sum[1];
        b[k].llen[0]^=b[k].llen[1]^=b[k].llen[0]^=b[k].llen[1];
        b[k].rlen[0]^=b[k].rlen[1]^=b[k].rlen[0]^=b[k].rlen[1];
        b[k].mxlen[0]^=b[k].mxlen[1]^=b[k].mxlen[0]^=b[k].mxlen[1];
    }
    else
    {
        pushdown(k);
        int mid=(b[k].ll+b[k].rr)>>1;
        if(l>mid)rev(k<<1|1,l,r);
        else if(r<=mid)rev(k<<1,l,r);
        else rev(k<<1,l,mid),rev(k<<1|1,mid+1,r);
        update(k);
    }
}
void pushdown(int k)
{
    if(b[k].laz==2)
    rev(k<<1,b[k<<1].ll,b[k<<1].rr),rev(k<<1|1,b[k<<1|1].ll,b[k<<1|1].rr);
    else if(b[k].laz>=0)
    mdf(k<<1,b[k<<1].ll,b[k<<1].rr,b[k].laz),mdf(k<<1|1,b[k<<1|1].ll,b[k<<1|1].rr,b[k].laz);
    b[k].laz=-1;
}
int qsum(int k,int l,int r)
{
    if(b[k].ll==l&&b[k].rr==r)
    return b[k].sum[1];
    else
    {
        pushdown(k);
        int mid=(b[k].ll+b[k].rr)>>1;
        if(l>mid)return qsum(k<<1|1,l,r);
        else if(r<=mid)return qsum(k<<1,l,r);
        else return qsum(k<<1,l,mid)+qsum(k<<1|1,mid+1,r);
    }
}
int qmax(int k,int l,int r)
{
    if(b[k].ll==l&&b[k].rr==r)
    return b[k].mxlen[1];
    else
    {
        pushdown(k);
        int mid=(b[k].ll+b[k].rr)>>1;
        if(l>mid)return qmax(k<<1|1,l,r);
        else if(r<=mid)return qmax(k<<1,l,r);
        else 
        {
            int ret1=max(qmax(k<<1,l,mid),qmax(k<<1|1,mid+1,r)),
            ret2=min(b[k<<1|1].llen[1],r-mid)+min(b[k<<1].rlen[1],mid-l+1);
            return max(ret1,ret2);
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=0;i<n;i++)
    scanf("%d",&a[i]);
    build(1,0,n-1);
    while(m--)
    {
        scanf("%d%d%d",&op,&x,&y);
        if(op==2)rev(1,x,y);
        else if(op==3)
        printf("%d\n",qsum(1,x,y));
        else if(op==4)
        printf("%d\n",qmax(1,x,y));
        else mdf(1,x,y,op);
    }
}