1858: [Scoi2010]序列操作
Time Limit: 10 Sec Memory Limit: 64 MB线段树,对于每个区间需要分别维护左右和中间的1和0连续个数,并在op=4时特殊处理一下。
Description
lxhgww最近收到了一个01序列,序列里面包含了n个数,这些数要么是0,要么是1,现在对于这个序列有五种变换操作和询问操作:
0 a b 把[a, b]区间内的所有数全变成0;
1 a b 把[a, b]区间内的所有数全变成1;
2 a b 把[a,b]区间内的所有数全部取反,也就是说把所有的0变成1,把所有的1变成0;
3 a b 询问[a, b]区间内总共有多少个1;
4 a b 询问[a, b]区间内最多有多少个连续的1。
对于每一种询问操作,lxhgww都需要给出回答,聪明的程序员们,你们能帮助他吗?
Input
输入数据第一行包括2个数,n和m,分别表示序列的长度和操作数目 第二行包括n个数,表示序列的初始状态 接下来m行,每行3个数,op, a, b,(0<=op<=4,0<=a<=b<n)表示对于区间[a, b]执行标号为op的操作。
Output
对于每一个询问操作,输出一行,包括1个数,表示其对应的答案
Sample Input
10 10
0 0 0 1 1 0 1 0 1 1
1 0 2
3 0 5
2 2 2
4 0 4
0 3 6
2 3 7
4 2 8
1 0 5
0 5 6
3 3 9
0 0 0 1 1 0 1 0 1 1
1 0 2
3 0 5
2 2 2
4 0 4
0 3 6
2 3 7
4 2 8
1 0 5
0 5 6
3 3 9
Sample Output
5
2
6
5
2
6
5
HINT
对于30%的数据,1<=n, m<=1000
对于100%的数据,1<=n, m<=100000
Source
#include<cstdio> #define M 100010 inline int max(int a,int b){return a>b?a:b;} struct tree{int l,r,s,lazy,sum,ml1,mr1,mm1,ml0,mr0,mm0;}tr[M*6]; int a[M],n,m,x,y,v,op; inline int read() { int tmp=0; char ch=getchar(); while(ch<'0'||ch>'9') ch=getchar(); while(ch>='0'&&ch<='9'){tmp=tmp*10+ch-'0';ch=getchar();} return tmp; } inline void pu(int p) { int p1=p<<1,p2=p<<1|1; tr[p].sum=tr[p1].sum+tr[p2].sum; tr[p].mm1=tr[p1].mr1+tr[p2].ml1; tr[p].mm0=tr[p1].mr0+tr[p2].ml0; if(tr[p1].mr1==tr[p1].s) tr[p].ml1=tr[p1].mr1+tr[p2].ml1; else tr[p].ml1=tr[p1].ml1; if(tr[p1].mr0==tr[p1].s) tr[p].ml0=tr[p1].mr0+tr[p2].ml0; else tr[p].ml0=tr[p1].ml0; if(tr[p2].mr1==tr[p2].s) tr[p].mr1=tr[p2].mr1+tr[p1].mr1; else tr[p].mr1=tr[p2].mr1; if(tr[p2].mr0==tr[p2].s) tr[p].mr0=tr[p2].mr0+tr[p1].mr0; else tr[p].mr0=tr[p2].mr0; tr[p].mm1=max(tr[p].mm1,max(tr[p1].mm1,tr[p2].mm1)); tr[p].mm0=max(tr[p].mm0,max(tr[p1].mm0,tr[p2].mm0)); } inline void change(int p) { int t1=tr[p].ml0,t2=tr[p].mr0,t3=tr[p].mm0; tr[p].ml0=tr[p].ml1; tr[p].ml1=t1; tr[p].mr0=tr[p].mr1; tr[p].mr1=t2; tr[p].mm0=tr[p].mm1; tr[p].mm1=t3; } inline void make(int l,int r,int p) { tr[p].l=l,tr[p].r=r,tr[p].s=tr[p].r-tr[p].l+1,tr[p].lazy=-1; if(l==r) { tr[p].sum=a[l]; tr[p].ml1=tr[p].mm1=tr[p].mr1=a[l]; tr[p].ml0=tr[p].mm0=tr[p].mr0=a[l]^1; return; } int mid=(l+r)>>1; make(l,mid,p<<1); make(mid+1,r,p<<1|1); pu(p); } inline void pd(int p) { int lz=tr[p].lazy,p1=p<<1,p2=p<<1|1; if(lz==0) { tr[p1].lazy=tr[p2].lazy=0; tr[p1].sum=0; tr[p2].sum=0; tr[p1].ml1=tr[p1].mr1=tr[p1].mm1=0; tr[p2].ml1=tr[p2].mr1=tr[p2].mm1=0; tr[p1].ml0=tr[p1].mr0=tr[p1].mm0=tr[p1].s; tr[p2].ml0=tr[p2].mr0=tr[p2].mm0=tr[p2].s; } else if(lz==1) { tr[p1].lazy=tr[p2].lazy=1; tr[p1].sum=tr[p1].s; tr[p2].sum=tr[p2].s; tr[p1].ml1=tr[p1].mr1=tr[p1].mm1=tr[p1].s; tr[p2].ml1=tr[p2].mr1=tr[p2].mm1=tr[p2].s; tr[p1].ml0=tr[p1].mr0=tr[p1].mm0=0; tr[p2].ml0=tr[p2].mr0=tr[p2].mm0=0; } else if(lz==2) { if(tr[p1].lazy==1) tr[p1].lazy=0; else if(tr[p1].lazy==0) tr[p1].lazy=1; else if(tr[p1].lazy==2) tr[p1].lazy=-1; else tr[p1].lazy=2; if(tr[p2].lazy==1) tr[p2].lazy=0; else if(tr[p2].lazy==0) tr[p2].lazy=1; else if(tr[p2].lazy==2) tr[p2].lazy=-1; else tr[p2].lazy=2; tr[p1].sum=tr[p1].s-tr[p1].sum; tr[p2].sum=tr[p2].s-tr[p2].sum; change(p1);change(p2); } tr[p].lazy=-1; } inline int find1(int l,int r,int p) { pd(p); if(tr[p].l==l&&tr[p].r==r) return tr[p].sum; int mid=(tr[p].l+tr[p].r)>>1; if(mid>=r) return find1(l,r,p<<1); else if(mid<l) return find1(l,r,p<<1|1); else return find1(l,mid,p<<1)+find1(mid+1,r,p<<1|1); } inline int findl(int l,int r,int p) { pd(p); if(tr[p].ml1+l>r) return r-l+1; else return tr[p].ml1; } inline int findr(int l,int r,int p) { pd(p); if(l+tr[p].mr1>r) return r-l+1; else return tr[p].mr1; } inline int find2(int l,int r,int p) { pd(p); if(tr[p].l==l&&tr[p].r==r) return max(tr[p].ml1,max(tr[p].mm1,tr[p].mr1)); int mid=(tr[p].l+tr[p].r)>>1; if(mid>=r) return find2(l,r,p<<1); else if(mid<l) return find2(l,r,p<<1|1); else return max(max(find2(l,mid,p<<1),find2(mid+1,r,p<<1|1)),findl(mid+1,r,p<<1|1)+findr(l,mid,p<<1)); } inline void xg(int l,int r,int c,int p) { pd(p); if(tr[p].l==l&&tr[p].r==r) { tr[p].lazy=c; if(c==0) { tr[p].ml1=tr[p].mr1=tr[p].mm1=0; tr[p].ml0=tr[p].mr0=tr[p].mm0=tr[p].s; tr[p].sum=0; } else if(c==1) { tr[p].ml0=tr[p].mr0=tr[p].mm0=0; tr[p].ml1=tr[p].mr1=tr[p].mm1=tr[p].s; tr[p].sum=tr[p].s; } else { tr[p].sum=tr[p].s-tr[p].sum; change(p); } return; } int mid=(tr[p].l+tr[p].r)>>1; if(mid>=r) xg(l,r,c,p<<1); else if(mid<l) xg(l,r,c,p<<1|1); else { xg(l,mid,c,p<<1); xg(mid+1,r,c,p<<1|1); } pu(p); } int main() { n=read(),m=read(); for(int i=1;i<=n;i++) a[i]=read(); make(1,n,1); for(int i=0;i<m;i++) { op=read();x=read();y=read(); if(op<3) xg(x+1,y+1,op,1); else if(op==3) printf("%d ",find1(x+1,y+1,1)); else printf("%d ",find2(x+1,y+1,1)); } return 0; }