0和1,分别将区间置0和置1,区间替换。
2,区间翻转,只需要注意线段树中的sum、lsum、rsum、msum的值就好
3,区间求和,得到几则有几个1
4,区间合并,注意在查询的时候合并。
线段树综合好题,基本上所有的有关的操作都糅合进去了。
奇迹般的1Y了!
#include<iostream>
#include<cstdio>
using namespace std;
#define maxn 100010
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
struct Tree
{
int l,r;
int lazy,sum;
int lsum,msum,rsum;
}tree[maxn<<2];
int num[maxn];
void PushUp(int rt)
{
tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
tree[rt].lsum=tree[rt<<1].lsum;
tree[rt].rsum=tree[rt<<1|1].rsum;
if(tree[rt].lsum==(tree[rt<<1].r-tree[rt<<1].l+1))
tree[rt].lsum+=tree[rt<<1|1].lsum;
if(tree[rt].rsum==(tree[rt<<1|1].r-tree[rt<<1|1].l+1))
tree[rt].rsum+=tree[rt<<1].rsum;
tree[rt].msum=max(tree[rt<<1].rsum+tree[rt<<1|1].lsum,max(tree[rt<<1].msum,tree[rt<<1|1].msum));
}
void PushDown(int rt)
{
if(tree[rt].lazy!=-1){
tree[rt<<1].lazy=tree[rt<<1|1].lazy=tree[rt].lazy;
tree[rt<<1].msum=tree[rt<<1].lsum=tree[rt<<1].rsum= tree[rt].lazy?(tree[rt<<1].r-tree[rt<<1].l+1):0;
tree[rt<<1].sum=tree[rt].lazy?(tree[rt<<1].r-tree[rt<<1].l+1):0;
tree[rt<<1|1].msum=tree[rt<<1|1].lsum=tree[rt<<1|1].rsum= tree[rt].lazy?(tree[rt<<1|1].r-tree[rt<<1|1].l+1):0;
tree[rt<<1|1].sum=tree[rt].lazy?(tree[rt<<1|1].r-tree[rt<<1|1].l+1):0;
tree[rt].lazy=-1;
}
}
void build(int l,int r,int rt)
{
tree[rt].l=l;
tree[rt].r=r;
tree[rt].lazy=-1;
if(l==r) {
tree[rt].sum=num[l];
if(num[l]){
tree[rt].lsum=tree[rt].rsum=tree[rt].msum=1;
tree[rt].sum=1;
tree[rt].lazy=1;
}
else{
tree[rt].lsum=tree[rt].rsum=tree[rt].msum=0;
tree[rt].sum=0;
tree[rt].lazy=0;
}
return ;
}
int m=(l+r)>>1;
build(lson);
build(rson);
PushUp(rt);
}
void update_t(int x,int y,int val,int rt)
{
int l,r;
l=tree[rt].l;
r=tree[rt].r;
if(l==x&&r==y){
tree[rt].lazy=val;
tree[rt].sum=tree[rt].lazy?r-l+1:0;
tree[rt].lsum=tree[rt].rsum=tree[rt].msum=tree[rt].lazy?r-l+1:0;
return ;
}
PushDown(rt);
int m=(l+r)>>1;
if(x<=m) update_t(x,min(m,y),val,rt<<1);
if(y>m) update_t(max(m+1,x),y,val,rt<<1|1);
PushUp(rt);
}
void update_l(int x,int y,int rt)
{
int l,r;
l=tree[rt].l;
r=tree[rt].r;
if(l==x&&r==y&&(tree[rt].sum==0||tree[rt].sum==r-l+1)){
if(tree[rt].sum>0)
tree[rt].lsum=tree[rt].msum=tree[rt].rsum=tree[rt].sum=tree[rt].lazy=0;
else{
tree[rt].lsum=tree[rt].msum=tree[rt].rsum=tree[rt].sum=tree[rt].lazy=r-l+1;
tree[rt].lazy=1;
}
return ;
}
PushDown(rt);
int m=(l+r)>>1;
if(x<=m) update_l(x,min(m,y),rt<<1);
if(y>m) update_l(max(m+1,x),y,rt<<1|1);
PushUp(rt);
}
int query(int x,int y,int val,int rt)
{
int l,r;
l=tree[rt].l;
r=tree[rt].r;
if(l==x&&r==y){
if(val) return tree[rt].msum;
else return tree[rt].sum;
}
PushDown(rt);
int m=(l+r)>>1;
if(y<=m) return query(x,y,val,rt<<1);
else if(x>m) return query(x,y,val,rt<<1|1);
else{
if(val){
return max(max(query(x,m,val,rt<<1),query(m+1,y,val,rt<<1|1)),min(tree[rt<<1].rsum,tree[rt<<1].r-x+1)+min(tree[rt<<1|1].lsum,y-tree[rt<<1|1].l+1));
}
else return query(x,m,val,rt<<1)+query(m+1,y,val,rt<<1|1);
}
}
int main()
{
int T,n,m;
scanf("%d",&T);
while(T--){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++)
scanf("%d",&num[i]);
build(1,n,1);
while(m--){
int op,a,b;
scanf("%d%d%d",&op,&a,&b);
a++;b++;
if(op==0)
update_t(a,b,0,1);
else if(op==1)
update_t(a,b,1,1);
else if(op==2)
update_l(a,b,1);
else if(op==3)
printf("%d\n",query(a,b,0,1));
else
printf("%d\n",query(a,b,1,1));
}
}
return 0;
}