这个题算是把线段树的基本操作融合在了一起,0和1是区间覆盖,2是区间异或,3是区间求和,4我们需要区间合并。
思维难度并不大,关键就是我们要把每一个操作做好。
我们一个一个分析操作
对于0,1,用一个标记cover解决
对于2,用一个标记rev解决,但是cover一个区间的时候,显然rev标记要清除。
对于3,在节点设置一个sum,代表1的个数,进行区间求和就行了。
对于4,我们设置most,lmost,rmost,分别记录区间最大连续1的个数,包括最左边的最大连续1的个数,包括最右边最大连续1的个数,
在pushup的时候我们就可以进行区间合并,但是这个题还有一个区间异或,所以我们设置tmost,tlmost,trmost,分别代表最长连续的0的个数。
这样我们在进行区间异或的时候,就可以直接进行这三个数的交换。
线段树功能: updata:区间求和,区间异或,区间合并 query:区间求和,区间合并
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define lc l,m,index<<1
#define rc m+1,r,index<<1|1
#define N 100005
struct node
{
int cover;
int rev;
int sum;
int most;
int lmost,rmost;
int tmost;
int tlmost,trmost;
}seg[N<<2];
int n,q;
void fcover(int flag,int l,int r,int index)
{
seg[index].cover=flag;
seg[index].rev=0;
if(flag==1)//覆盖成0
{
seg[index].sum=seg[index].most=seg[index].lmost=seg[index].rmost=0;
seg[index].tmost=seg[index].tlmost=seg[index].trmost=r-l+1;
}
else//覆盖为1
{
seg[index].sum=seg[index].most=seg[index].lmost=seg[index].rmost=r-l+1;
seg[index].tmost=seg[index].tlmost=seg[index].trmost=0;
}
}
void frev(int l,int r,int index)
{
seg[index].rev^=1;
seg[index].sum=r-l+1-seg[index].sum;
swap(seg[index].most,seg[index].tmost);
swap(seg[index].lmost,seg[index].tlmost);
swap(seg[index].rmost,seg[index].trmost);
}
void pushup(int l,int r,int index)
{
int m=(l+r)>>1;
node& father=seg[index];
node& lson=seg[index<<1];
node& rson=seg[index<<1|1];
father.sum=lson.sum+rson.sum;
father.most=max(lson.most,rson.most);
father.most=max(father.most,lson.rmost+rson.lmost);
father.lmost=lson.lmost;
father.rmost=rson.rmost;
if(lson.lmost==m-l+1)father.lmost+=rson.lmost;
if(rson.rmost==r-m)father.rmost+=lson.rmost;
father.tmost=max(lson.tmost,rson.tmost);
father.tmost=max(father.tmost,lson.trmost+rson.tlmost);
father.tlmost=lson.tlmost;
father.trmost=rson.trmost;
if(lson.tlmost==m-l+1)father.tlmost+=rson.tlmost;
if(rson.trmost==r-m)father.trmost+=lson.trmost;
}
void pushdown(int l,int r,int index)
{
int m=(l+r)>>1;
node& father=seg[index];
node& lson=seg[index<<1];
node& rson=seg[index<<1|1];
if(father.cover)
{
fcover(father.cover,lc);
fcover(father.cover,rc);
father.cover=0;
}
if(father.rev)
{
frev(lc);
frev(rc);
father.rev=0;
}
}
void build(int l,int r,int index)
{
seg[index].cover=seg[index].rev=0;
int m=(l+r)>>1;
if(l==r)
{
scanf("%d",&seg[index].sum);
seg[index].most=seg[index].lmost=seg[index].rmost=seg[index].sum;
seg[index].tmost=seg[index].tlmost=seg[index].trmost=seg[index].sum^1;
return;
}
build(lc);
build(rc);
pushup(l,r,index);
}
void updata(int op,int L,int R,int l,int r,int index)
{
int m=(l+r)>>1;
if(L==l&&R==r)
{
if(op==0)fcover(1,l,r,index);
else if(op==1)fcover(2,l,r,index);
else frev(l,r,index);
return;
}
pushdown(l,r,index);
if(R<=m)updata(op,L,R,lc);
else if(L>m)updata(op,L,R,rc);
else
{
updata(op,L,m,lc);
updata(op,m+1,R,rc);
}
pushup(l,r,index);
}
int query(int op,int L,int R,int l,int r,int index)
{
int m=(l+r)>>1;
if(L==l&&R==r)
{
if(op==3)return seg[index].sum;
else return seg[index].most;
}
pushdown(l,r,index);
if(R<=m)return query(op,L,R,lc);
else if(L>m)return query(op,L,R,rc);
else
{
if(op==3)return query(op,L,m,lc)+query(op,m+1,R,rc);
else
{
int ret;
ret=max(query(op,L,m,lc),query(op,m+1,R,rc));
ret=max(ret,min(R,seg[index<<1|1].lmost+m)-max(L,m-seg[index<<1].rmost+1)+1);
return ret;
}
}
}
int main()
{
int t,op,a,b;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&q);
build(0,n-1,1);
while(q--)
{
scanf("%d%d%d",&op,&a,&b);
if(op<=2)updata(op,a,b,0,n-1,1);
else printf("%d\n",query(op,a,b,0,n-1,1));
}
}
return 0;
}