很麻烦的一道线段树。。。
#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;
}