bzoj 1858: [Scoi2010]序列操作
Contents
1858: [Scoi2010]序列操作
Time Limit: 10 Sec Memory Limit: 64 MB
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
线段树，对于每个区间需要分别维护左右和中间的1和0连续个数，并在op=4时特殊处理一下。
#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\n",find1(x+1,y+1,1));
else printf("%d\n",find2(x+1,y+1,1));
}
return 0;
}
发表评论