题意:5种操作
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
题解
线段树上维护 \(cnt[0/1]\) 表示 \(0/1\) 的个数, \(lmax[0/1],rmax[0/1],tmax[0/1]\) 表示前缀最大个数,后缀最大个数,总的最大个数。
#include<bits/stdc++.h>
#define REP(i,a,b) for(int i(a);i<=(b);++i)
using namespace std;
typedef long long ll;
inline int read(){char c;int w;
while(!isdigit(c=getchar()));w=c&15;
while(isdigit(c=getchar()))w=w*10+(c&15);return w;
}
inline char smax(int&x,const int&y){return x<y?x=y,1:0;}
inline char smin(int&x,const int&y){return x>y?x=y,1:0;}
const int N=1e5+5;
struct data{int len,cnt,lmax,rmax,tmax;};
inline data merge(const data&a,const data&b){
data c;c.cnt=a.cnt+b.cnt;c.len=a.len+b.len;
c.lmax=a.cnt==a.len?a.lmax+b.lmax:a.lmax;
c.rmax=b.cnt==b.len?a.rmax+b.rmax:b.rmax;
c.tmax=max(a.tmax,b.tmax);
smax(c.tmax,a.rmax+b.lmax);
return c;
}
struct node{data c[2];bool rev;int len,set;}t[N<<2];
#define ls o<<1
#define rs o<<1|1
inline void change(int o,int w){
if(!o)return;t[o].rev=0,t[o].set=w;int&len=t[o].len;
t[o].c[w]=(data){len,len,len,len,len};
t[o].c[!w]=(data){len,0,0,0,0};
}
inline void rever(int o){
if(!o)return;
if(~t[o].set)change(o,!t[o].set);
else t[o].rev^=1,swap(t[o].c[0],t[o].c[1]);
}
#define pushup() \
t[o].c[0]=merge(t[ls].c[0],t[rs].c[0]),\
t[o].c[1]=merge(t[ls].c[1],t[rs].c[1])
#define pushdown() \
if(~t[o].set)change(ls,t[o].set),change(rs,t[o].set),t[o].set=-1;\
if(t[o].rev)rever(ls),rever(rs),t[o].rev=0;
inline data ask(int o,int l,int r,int x,int y){
if(x<=l&&r<=y)return t[o].c[1];
int mid=l+r>>1;pushdown();
if(y<=mid)return ask(ls,l,mid,x,y);
if(x>mid)return ask(rs,mid+1,r,x,y);
return merge(ask(ls,l,mid,x,y),ask(rs,mid+1,r,x,y));
}
inline void update(int o,int l,int r,int x,int y){
if(x<=l&&r<=y){rever(o);return;}
int mid=l+r>>1;pushdown();
if(x<=mid)update(ls,l,mid,x,y);
if(y>mid)update(rs,mid+1,r,x,y);
pushup();
}
inline void update(int o,int l,int r,int x,int y,int z){
if(x<=l&&r<=y){change(o,z);return;}
int mid=l+r>>1;pushdown();
if(x<=mid)update(ls,l,mid,x,y,z);
if(y>mid)update(rs,mid+1,r,x,y,z);
pushup();
}
int n,m,a[N];
inline void build(int o,int l,int r){
t[o].set=-1;t[o].rev=0;t[o].len=r-l+1;
if(l==r){change(o,a[l]);return;}
int mid=l+r>>1;build(ls,l,mid),build(rs,mid+1,r);
pushup();
}
int main(){
n=read();m=read();
#define all 1,1,n
REP(i,1,n)a[i]=read();build(all);
while(m--){
int op=read(),a=read()+1,b=read()+1;
switch(op){
case 0:update(all,a,b,0);break;
case 1:update(all,a,b,1);break;
case 2:update(all,a,b);break;
case 3:printf("%d\n",ask(all,a,b).cnt);break;
case 4:printf("%d\n",ask(all,a,b).tmax);break;
}
// REP(i,1,n)printf("%d ",ask(all,i,i).cnt);puts("");
}
return 0;
}