bzoj1858: [Scoi2010]序列操作

传送门
线段树裸题。
每个节点维护0/1的个数，覆盖标记，取反标记，在左边/右边连续0/1的个数。
下传上传自己xjb乱搞一下就可以了。

#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
struct node{
    int l,r,l0,r0,l1,r1,mx0,mx1,sum0,sum1,rev,c,full;
}t[400005];
int n,m,f,x,y;
void rev(int k){
    swap(t[k].l0,t[k].l1);
    swap(t[k].r0,t[k].r1);
    swap(t[k].mx0,t[k].mx1);
    swap(t[k].sum0,t[k].sum1);
    if (t[k].full!=-1) t[k].full^=1;
}
void color(int k,int v){
    t[k].rev=0;
    int s=t[k].r-t[k].l+1;
    t[k].mx0=t[k].l0=t[k].r0=t[k].sum0=(v==0)?s:0;
    t[k].mx1=t[k].l1=t[k].r1=t[k].sum1=(v!=0)?s:0;
    t[k].full=v;
}
node merge(node a,node b){
    node tmp;
    tmp.l=a.l; tmp.r=b.r;
    tmp.rev=0; tmp.c=-1;
    tmp.l0=a.l0; tmp.l1=a.l1;
    tmp.r0=b.r0; tmp.r1=b.r1;
    tmp.mx0=max(a.mx0,max(b.mx0,a.r0+b.l0));
    tmp.mx1=max(a.mx1,max(b.mx1,a.r1+b.l1));
    tmp.sum0=a.sum0+b.sum0;
    tmp.sum1=a.sum1+b.sum1;
    if (a.full==0) tmp.l0=a.mx0+b.l0;
        else if (a.full==1) tmp.l1=a.mx1+b.l1;
    if (b.full==0) tmp.r0=b.mx0+a.r0;
        else if (b.full==1) tmp.r1=b.mx1+a.r1;
    tmp.full=(a.full==b.full)?a.full:-1;
    return tmp;
}
void pushdown(int k){
    if (t[k].l==t[k].r) return;
    if (t[k].c!=-1){
        t[k<<1].c=t[k<<1|1].c=t[k].c;
        color(k<<1,t[k].c);
        color(k<<1|1,t[k].c);
        t[k].c=-1;
    }
    if (t[k].rev){
        t[k<<1].rev^=1;
        t[k<<1|1].rev^=1;
        rev(k<<1);
        rev(k<<1|1);
        t[k].rev=0;
    }
}
void build(int k,int l,int r){
    t[k].l=l; t[k].r=r; t[k].c=-1;
    if (l==r){
        scanf("%d",&t[k].full);
        t[k].l1=t[k].r1=t[k].mx1=t[k].sum1=t[k].full;
        t[k].l0=t[k].r0=t[k].mx0=t[k].sum0=1-t[k].full;
        return;
    }
    int mid=(l+r)>>1;
    build(k<<1,l,mid);
    build(k<<1|1,mid+1,r);
    t[k]=merge(t[k<<1],t[k<<1|1]);
}
void change(int k,int x,int y,int v){
    pushdown(k);
    int l=t[k].l,r=t[k].r;
    if (x==l&&y==r){
        t[k].c=v;
        color(k,v);
        return;
    }
    int mid=(l+r)>>1;
    if (y<=mid) change(k<<1,x,y,v);
    else if (x>mid) change(k<<1|1,x,y,v);
    else change(k<<1,x,mid,v),change(k<<1|1,mid+1,y,v);
    t[k]=merge(t[k<<1],t[k<<1|1]);
}
void rever(int k,int x,int y){
    pushdown(k);
    int l=t[k].l,r=t[k].r;
    if (x==l&&y==r){
        t[k].rev=1;
        rev(k);
        return;
    }
    int mid=(l+r)>>1;
    if (y<=mid) rever(k<<1,x,y);
    else if (x>mid) rever(k<<1|1,x,y);
    else rever(k<<1,x,mid),rever(k<<1|1,mid+1,y);
    t[k]=merge(t[k<<1],t[k<<1|1]);
}
node askmax(int k,int x,int y){
    pushdown(k);
    int l=t[k].l,r=t[k].r;
    if (l==x&&y==r) return t[k];
    int mid=(l+r)>>1;
    if (y<=mid) return askmax(k<<1,x,y);
    else if (x>mid) return askmax(k<<1|1,x,y);
    else return merge(askmax(k<<1,x,mid),askmax(k<<1|1,mid+1,y));
}
int asksum(int k,int x,int y){
    pushdown(k);
    int l=t[k].l,r=t[k].r;
    if (l==x&&y==r) return t[k].sum1;
    int mid=(l+r)>>1;
    if (y<=mid) return asksum(k<<1,x,y);
    else if (x>mid) return asksum(k<<1|1,x,y);
    else return asksum(k<<1,x,mid)+asksum(k<<1|1,mid+1,y);
}
int main(){
    scanf("%d%d",&n,&m);
    build(1,1,n);
    for (int i=1;i<=m;i++){
        scanf("%d%d%d",&f,&x,&y);
        x++; y++;
        if (f<2) change(1,x,y,f);
        else if (f==2) rever(1,x,y);
        else if (f==3) printf("%d\n",asksum(1,x,y));
        else printf("%d\n",askmax(1,x,y).mx1);
    }
}