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都需要给出回答,聪明的程序员们,你们能帮助他吗?
这道题我还能讲什么呢?相信大家都是一眼秒算法,线段树嘛。但这是道码农题啊,需要维护的东西太多了。关键是两个lazy标记的下放,要看哪个先再更新,就要新开个标记来记录它的顺序,之后就是套路了,具体看代码。
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
using namespace std;
struct trnode
{
int l,r,lc,rc;
int c1,c2,c3;int k1,k2,k3,k4;
int lazy1,lazy2,t1,t2;
trnode(){lazy1=-1;lazy2=0;}
//c1表示区间内1的个数 c2表示区间内最多有连续1的个数 c3表示示区间内最多有连续0的个数
//k1表示左端点开始连续1的个数 k2表示右端点开始连续1的个数 k3表示左端点开始连续0的个数 k4表示右端点开始连续0的个数
}tr[210000];int trlen;
int a[110000];
inline void myswap(int now,int t)
{
tr[now].c1=tr[now].r-tr[now].l+1-tr[now].c1;
swap(tr[now].c2,tr[now].c3);swap(tr[now].k1,tr[now].k3);swap(tr[now].k2,tr[now].k4);
tr[now].lazy2=1-tr[now].lazy2;tr[now].t2=t;
}
inline void xg(int now,int k,int t)
{
tr[now].lazy1=k;tr[now].lazy2=0;tr[now].t1=t;
if(k==1)
{
tr[now].c1=tr[now].c2=tr[now].r-tr[now].l+1;tr[now].c3=0;
tr[now].k1=tr[now].k2=tr[now].r-tr[now].l+1;tr[now].k3=tr[now].k4=0;
}
if(k==0)
{
tr[now].c1=tr[now].c2=0;tr[now].c3=tr[now].r-tr[now].l+1;
tr[now].k1=tr[now].k2=0;tr[now].k3=tr[now].k4=tr[now].r-tr[now].l+1;
}
}
inline void update1(int now)
{
int lc=tr[now].lc,rc=tr[now].rc;
tr[now].c1=tr[lc].c1+tr[rc].c1;
tr[now].c2=max(max(tr[lc].c2,tr[rc].c2),tr[lc].k2+tr[rc].k1);
tr[now].c3=max(max(tr[lc].c3,tr[rc].c3),tr[lc].k4+tr[rc].k3);
tr[now].k1=tr[lc].k1;if(tr[lc].k1==tr[lc].r-tr[lc].l+1)tr[now].k1+=tr[rc].k1;
tr[now].k2=tr[rc].k2;if(tr[rc].k2==tr[rc].r-tr[rc].l+1)tr[now].k2+=tr[lc].k2;
tr[now].k3=tr[lc].k3;if(tr[lc].k3==tr[lc].r-tr[lc].l+1)tr[now].k3+=tr[rc].k3;
tr[now].k4=tr[rc].k4;if(tr[rc].k4==tr[rc].r-tr[rc].l+1)tr[now].k4+=tr[lc].k4;
}
inline void update2(int now,int t)
{
int lc=tr[now].lc,rc=tr[now].rc;
xg(lc,tr[now].lazy1,t);
xg(rc,tr[now].lazy1,t);
tr[now].lazy1=-1;
}
inline void update3(int now,int t)
{
int lc=tr[now].lc,rc=tr[now].rc;
myswap(lc,t);
myswap(rc,t);
tr[now].lazy2=0;
}
inline void update(int now)
{
int lc=tr[now].lc,rc=tr[now].rc;
if(tr[now].t1>tr[now].t2)
{
if(tr[now].lazy2!=0)update3(now,tr[now].t2);
if(tr[now].lazy1!=-1)update2(now,tr[now].t1);
}
else
{
if(tr[now].lazy1!=-1)update2(now,tr[now].t1);
if(tr[now].lazy2!=0)update3(now,tr[now].t2);
}
}
void bt(int l,int r)
{
trlen++;int now=trlen;
tr[now].l=l;tr[now].r=r;
tr[now].lc=tr[now].rc=-1;
if(l==r)
{
tr[now].c1=tr[now].c2=a[l];tr[now].c3=(1-a[l]);
tr[now].k1=tr[now].k2=a[l];tr[now].k3=tr[now].k4=(1-a[l]);
}
else
{
int mid=(l+r)/2;
tr[now].lc=trlen+1;bt(l,mid);
tr[now].rc=trlen+1;bt(mid+1,r);
update1(now);
}
}
void change(int now,int l,int r,int k,int t)
{
if(tr[now].l==l && tr[now].r==r){xg(now,k,t);return ;}
update(now);
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(r<=mid)change(lc,l,r,k,t);
else if(mid+1<=l)change(rc,l,r,k,t);
else change(lc,l,mid,k,t),change(rc,mid+1,r,k,t);
update1(now);
}
void getf(int now,int l,int r,int t)
{
if(tr[now].l==l && tr[now].r==r){myswap(now,t);return ;}
update(now);
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(r<=mid)getf(lc,l,r,t);
else if(mid+1<=l)getf(rc,l,r,t);
else getf(lc,l,mid,t),getf(rc,mid+1,r,t);
update1(now);
}
int solve1(int now,int l,int r)
{
if(tr[now].l==l && tr[now].r==r)return tr[now].c1;
update(now);
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(r<=mid)return solve1(lc,l,r);
else if(mid+1<=l)return solve1(rc,l,r);
else return solve1(lc,l,mid)+solve1(rc,mid+1,r);
}
int solve2(int now,int l,int r)
{
if(tr[now].l==l && tr[now].r==r)return tr[now].c2;
update(now);
int lc=tr[now].lc,rc=tr[now].rc,mid=(tr[now].l+tr[now].r)/2;
if(r<=mid)return solve2(lc,l,r);
else if(mid+1<=l)return solve2(rc,l,r);
else return max(max(solve2(lc,l,mid),solve2(rc,mid+1,r)),min(tr[lc].r-l+1,tr[lc].k2)+min(r-tr[rc].l+1,tr[rc].k1));
}
int main()
{
int n,m;
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)scanf("%d",&a[i]);
bt(1,n);
for(int i=1;i<=m;i++)
{
int k,x,y;
scanf("%d%d%d",&k,&x,&y);x++;y++;
if(k==0)change(1,x,y,0,i);
if(k==1)change(1,x,y,1,i);
if(k==2)getf(1,x,y,i);
if(k==3)printf("%d\n",solve1(1,x,y));
if(k==4)printf("%d\n",solve2(1,x,y));
}
return 0;
}