题意: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#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;}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|1inline 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;}