SCOI2010序列操作--线段树迷之维护

硬刚了一晚上终于刚完了，发篇博客纪念一下。

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn=100005;
struct Seg_Tree{
int sum[maxn<<2],Max[maxn<<2][2],fuz[maxn<<2];
int Maxl[maxn<<2][2],Maxr[maxn<<2][2];
bool rev[maxn<<2];
void push_down(int p,int u,int v,int l,int r,int mid){
if(fuz[p]>=0){
int k=fuz[p];
sum[u]=(mid-l+1)*k;
sum[v]=(r-mid)*k;
Max[u][k]=Maxl[u][k]=Maxr[u][k]=mid-l+1;
Max[u][k^1]=Maxl[u][k^1]=Maxr[u][k^1]=0;
Max[v][k]=Maxl[v][k]=Maxr[v][k]=r-mid;
Max[v][k^1]=Maxl[v][k^1]=Maxr[v][k^1]=0;
fuz[u]=fuz[v]=fuz[p];
fuz[p]=-1,rev[p]=0;
}else if(rev[p]){
swap(Max[u][0],Max[u][1]);
swap(Maxl[u][0],Maxl[u][1]);
swap(Maxr[u][0],Maxr[u][1]);
swap(Max[v][0],Max[v][1]);
swap(Maxl[v][0],Maxl[v][1]);
swap(Maxr[v][0],Maxr[v][1]);
sum[u]=mid-l+1-sum[u];
sum[v]=r-mid-sum[v];
rev[u]^=1,rev[v]^=1;
if(fuz[u]>=0)fuz[u]^=1;
if(fuz[v]>=0)fuz[v]^=1;
rev[p]=0;
}
}
void push_up(int p,int u,int v,int l,int r,int mid){
Maxl[p][0]=Maxl[u][0];
if(Maxl[p][0]==mid-l+1)Maxl[p][0]+=Maxl[v][0];
Maxl[p][1]=Maxl[u][1];
if(Maxl[p][1]==mid-l+1)Maxl[p][1]+=Maxl[v][1];
Maxr[p][0]=Maxr[v][0];
if(Maxr[p][0]==r-mid)Maxr[p][0]+=Maxr[u][0];
Maxr[p][1]=Maxr[v][1];
if(Maxr[p][1]==r-mid)Maxr[p][1]+=Maxr[u][1];
Max[p][0]=max(Maxr[u][0]+Maxl[v][0],max(Max[u][0],Max[v][0]));
Max[p][1]=max(Maxr[u][1]+Maxl[v][1],max(Max[u][1],Max[v][1]));
sum[p]=sum[u]+sum[v];
}
void create_tree(int p,int l,int r){
fuz[p]=-1,rev[p]=0;
if(l==r){
int k;
scanf("%d",&k);
sum[p]=k;
Maxl[p][k]=Maxr[p][k]=Max[p][k]=1;
Maxl[p][k^1]=Maxr[p][k^1]=Max[p][k^1]=0;
return;
}
int mid=(l+r)>>1,u=p<<1,v=u+1;
create_tree(u,l,mid);
create_tree(v,mid+1,r);
push_up(p,u,v,l,r,mid);
}
void update(int p,int l,int r,int L,int R,int val){
if((L<=l)&&(R>=r)){
fuz[p]=val,rev[p]=0,sum[p]=(r-l+1)*val;
Max[p][val]=Maxl[p][val]=Maxr[p][val]=r-l+1;
Max[p][val^1]=Maxl[p][val^1]=Maxr[p][val^1]=0;
return;
}
int mid=(l+r)>>1,u=p<<1,v=u+1;
push_down(p,u,v,l,r,mid);
if(L<=mid)update(u,l,mid,L,R,val);
if(R>mid)update(v,mid+1,r,L,R,val);
push_up(p,u,v,l,r,mid);
}
void revers(int p,int l,int r,int L,int R){
if((L<=l)&&(R>=r)){
swap(Max[p][0],Max[p][1]);
swap(Maxl[p][0],Maxl[p][1]);
swap(Maxr[p][0],Maxr[p][1]);
sum[p]=r-l+1-sum[p],rev[p]^=1;
if(fuz[p]>=0)fuz[p]^=1;
return;
}
int mid=(l+r)>>1,u=p<<1,v=u+1;
push_down(p,u,v,l,r,mid);
if(L<=mid)revers(u,l,mid,L,R);
if(R>mid)revers(v,mid+1,r,L,R);
push_up(p,u,v,l,r,mid);
}
int query_sum(int p,int l,int r,int L,int R){
if((L<=l)&&(R>=r))
return sum[p];
int mid=(l+r)>>1,u=p<<1,v=u+1,res=0;
push_down(p,u,v,l,r,mid);
if(L<=mid)res=query_sum(u,l,mid,L,R);
if(R>mid)res+=query_sum(v,mid+1,r,L,R);
push_up(p,u,v,l,r,mid);
return res;
}
int query(int p,int l,int r,int L,int R){
if((L<=l)&&(R>=r))
return Max[p][1];
int mid=(l+r)>>1,u=p<<1,v=u+1,res=0;
push_down(p,u,v,l,r,mid);
if(L<=mid)res=query(u,l,mid,L,R);
if(R>mid)res=max(res,query(v,mid+1,r,L,R));
if((L<=mid)&&(R>mid))
res=max(res,min(Maxr[u][1],mid-L+1)+min(Maxl[v][1],R-mid));
push_up(p,u,v,l,r,mid);
return res;
}
}T;
int main(){
int n,m,type,l,r;
scanf("%d%d",&n,&m);
T.create_tree(1,1,n);
while(m--){
scanf("%d%d%d",&type,&l,&r),l++,r++;
if(type==0)T.update(1,1,n,l,r,0);
else if(type==1)T.update(1,1,n,l,r,1);
else if(type==2)T.revers(1,1,n,l,r);
else if(type==3)printf("%d\n",T.query_sum(1,1,n,l,r));
else printf("%d\n",T.query(1,1,n,l,r));
}
return 0;
}