【BZOJ1858】序列操作(线段树)
【BZOJ1858】序列操作(线段树)
题面
题解
这题思路很简单,细节很烦,很码
维护区间翻转和区间赋值标记
当打到区间赋值标记时直接覆盖掉翻转标记
下放标记的时候先放赋值标记再放翻转标记
这样可以维护前4个操作
对于第5个操作
维护区间从左/右端点开始的最大连续\(0/1\)的个数
以及区间内的最大连续\(0/1\)的个数
做区间翻转的时候所有的关于\(0/1\)连续个数的计数全部要交换过来
其他的细节自己注意一下
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define ll long long
#define RG register
#define MAX 120000
#define lson (now<<1)
#define rson (now<<1|1)
inline int read()
{
RG int x=0,t=1;RG char ch=getchar();
while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
if(ch=='-')t=-1,ch=getchar();
while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
return x*t;
}
int n,m;
struct Node
{
int tag1,tag2;
int v,l,r;
int lg0,rg0,mg0;
int lg1,rg1,mg1;
void len0(){lg0=rg0=mg0=r-l+1;lg1=rg1=mg1=0;}
void len1(){lg1=rg1=mg1=r-l+1;lg0=rg0=mg0=0;}
void rev(){swap(lg0,lg1);swap(rg0,rg1);swap(mg0,mg1);}
void clear(){tag1=-1;tag2=v=l=r=lg0=rg0=mg0=lg1=rg1=mg1=0;}
}t[MAX<<2];
Node operator+(Node a,Node b)
{
Node c;c.tag1=-1;c.tag2=0;
c.v=a.v+b.v;c.l=a.l;c.r=b.r;
c.lg1=a.lg1;c.rg1=b.rg1;
if(a.mg1==a.r-a.l+1)c.lg1+=b.lg1;
if(b.mg1==b.r-b.l+1)c.rg1+=a.rg1;
c.mg1=max(a.mg1,b.mg1);
c.mg1=max(c.mg1,c.lg1);
c.mg1=max(c.mg1,c.rg1);
c.mg1=max(c.mg1,a.rg1+b.lg1);
c.lg0=a.lg0;c.rg0=b.rg0;
if(a.mg0==a.r-a.l+1)c.lg0+=b.lg0;
if(b.mg0==b.r-b.l+1)c.rg0+=a.rg0;
c.mg0=max(a.mg0,b.mg0);
c.mg0=max(c.mg0,c.lg0);
c.mg0=max(c.mg0,c.rg0);
c.mg0=max(c.mg0,a.rg0+b.lg0);
return c;
}
void Build(int now,int l,int r)
{
t[now].l=l;t[now].r=r;
t[now].tag1=-1;t[now].tag2=0;
if(l==r)
{
t[now].v=read();
if(t[now].v)t[now].len1();
else t[now].len0();
return;
}
int mid=(l+r)>>1;
Build(lson,l,mid);Build(rson,mid+1,r);
t[now]=t[lson]+t[rson];
}
void putrev(int now,int l,int r)
{
t[now].v=r-l+1-t[now].v;
t[now].rev();
t[now].tag2^=1;
}
void puteql(int now,int l,int r,int w)
{
if(w)t[now].len1(),t[now].v=r-l+1;
else t[now].len0(),t[now].v=0;
t[now].tag1=w;t[now].tag2=0;
}
void pushdown(int now,int l,int r)
{
int mid=(l+r)>>1;
if(t[now].tag1!=-1)
{
puteql(lson,l,mid,t[now].tag1);
puteql(rson,mid+1,r,t[now].tag1);
t[now].tag1=-1;
}
if(t[now].tag2)
{
putrev(lson,l,mid);
putrev(rson,mid+1,r);
t[now].tag2^=1;
}
}
void Modify_Eql(int now,int L,int R,int w)
{
if(L<=t[now].l&&t[now].r<=R){puteql(now,t[now].l,t[now].r,w);return;}
pushdown(now,t[now].l,t[now].r);
int mid=(t[now].l+t[now].r)>>1;
if(L<=mid)Modify_Eql(lson,L,R,w);
if(R>mid)Modify_Eql(rson,L,R,w);
t[now]=t[lson]+t[rson];
}
void Modify_rev(int now,int L,int R)
{
if(L<=t[now].l&&t[now].r<=R){putrev(now,t[now].l,t[now].r);return;}
pushdown(now,t[now].l,t[now].r);
int mid=(t[now].l+t[now].r)>>1;
if(L<=mid)Modify_rev(lson,L,R);
if(R>mid)Modify_rev(rson,L,R);
t[now]=t[lson]+t[rson];
}
int Query_Sum(int now,int L,int R)
{
if(L<=t[now].l&&t[now].r<=R)return t[now].v;
pushdown(now,t[now].l,t[now].r);
int mid=(t[now].l+t[now].r)>>1,ret=0;
if(L<=mid)ret+=Query_Sum(lson,L,R);
if(R>mid)ret+=Query_Sum(rson,L,R);
return ret;
}
Node Query_One(int now,int L,int R)
{
if(L==t[now].l&&t[now].r==R)return t[now];
pushdown(now,t[now].l,t[now].r);
int mid=(t[now].l+t[now].r)>>1;
if(R<=mid)return Query_One(lson,L,R);
if(L>mid)return Query_One(rson,L,R);
return Query_One(lson,L,mid)+Query_One(rson,mid+1,R);
}
int main()
{
n=read();m=read();
Build(1,1,n);
while(m--)
{
int opt=read(),l=read()+1,r=read()+1;
if(opt==0)Modify_Eql(1,l,r,0);
if(opt==1)Modify_Eql(1,l,r,1);
if(opt==2)Modify_rev(1,l,r);
if(opt==3)printf("%d\n",Query_Sum(1,l,r));
if(opt==4)printf("%d\n",Query_One(1,l,r).mg1);
}
return 0;
}