Luogu P2572 [SCOI2010]序列操作 线段树。。

时间:2021-11-10 16:40:41

咕咕了。。。于是借鉴了小粉兔的做法ORZ。。。


其实就是维护最大子段和的线段树,但上面又多了一些操作。。。。QWQ

维护8个信息:1/0的个数(sum),左/右边起1/0的最长长度(ls,rs),整段区间中1/0的连续最长长度(mx)。

于是对于各个操作,我们有了一些tag。。。

tg1[]是区间赋值标记,没有标记时为-1,有标记时为0或1;tg2[]是区间取反标记,没有标记时为 0,有标记时为1。

注意标记下传时要先传tg1[],再传tg2[],否则取反标记会被赋值标记覆盖

#include<cstdio>
#include<iostream>
#define R register int
#define ls (tr<<1)
#define rs (tr<<1|1)
const int N=;
using namespace std;
inline int g() {
R ret=,fix=; register char ch; while(!isdigit(ch=getchar())) fix=ch=='-'?-:fix;
do ret=ret*+(ch^); while(isdigit(ch=getchar())); return ret*fix;
}
int n,q,a[N];
struct node{
int sum0,sum1,ls0,ls1,rs0,rs1,mx0,mx1;
node(int s0=,int s1=,int ls0=,int ls1=,int rs0=,int rs1=,int mx0=,int mx1=):
sum0(s0),sum1(s1),ls0(ls0),ls1(ls1),rs0(rs0),rs1(rs1),mx0(mx0),mx1(mx1) {}
};
inline node upd(node l,node r) {
return node(l.sum0+r.sum0,l.sum1+r.sum1,
l.sum1?l.ls0:l.sum0+r.ls0,l.sum0?l.ls1:l.sum1+r.ls1,
r.sum1?r.rs0:r.sum0+l.rs0,r.sum0?r.rs1:r.sum1+l.rs1,
max(max(l.mx0,r.mx0),l.rs0+r.ls0),
max(max(l.mx1,r.mx1),l.rs1+r.ls1));
} node t[N]; int len[N],tg1[N],tg2[N];
inline void push(int tr,int typ) {
node& tmp=t[tr];
if(typ==) tg2[tr]=,tg1[tr]=,tmp=node(,len[tr],,len[tr],,len[tr],,len[tr]);
else if(typ==) tg2[tr]=,tg1[tr]=,tmp=node(len[tr],,len[tr],,len[tr],,len[tr],);
else if(typ==) tg2[tr]^=,swap(tmp.sum0,tmp.sum1),swap(tmp.ls0,tmp.ls1),swap(tmp.rs0,tmp.rs1),swap(tmp.mx0,tmp.mx1);
}
inline void spread(int tr) {
if(~tg1[tr]) push(ls,tg1[tr]),push(rs,tg1[tr]);
if(tg2[tr]) push(ls,),push(rs,);
tg1[tr]=-,tg2[tr]=;
}
inline void build(int tr,int l,int r) {
len[tr]=r-l+,tg1[tr]=-;
if(l==r) {R tmp=g(); t[tr]=node(tmp,tmp^,tmp,tmp^,tmp,tmp^,tmp,tmp^); return ;} R md=l+r>>;
build(ls,l,md),build(rs,md+,r); t[tr]=upd(t[ls],t[rs]);
}
inline void change(int tr,int l,int r,int LL,int RR,int d) {
if(LL<=l&&r<=RR) {push(tr,d); return ;} spread(tr); R md=l+r>>;
if(LL<=md) change(ls,l,md,LL,RR,d); if(RR>md) change(rs,md+,r,LL,RR,d);
t[tr]=upd(t[ls],t[rs]);
}
inline node query(int tr,int l,int r,int LL,int RR) {
if(LL<=l&&r<=RR) return t[tr]; spread(tr); R md=l+r>>; register node ret=node();
if(LL<=md) ret=query(ls,l,md,LL,RR); if(RR>md) ret=upd(ret,query(rs,md+,r,LL,RR)); return ret;
}
signed main() {
n=g(),q=g(); build(,,n);
for(R i=;i<=q;++i) {
R op=g(),l=g()+,r=g()+;
if(op<) change(,,n,l,r,op);
else {register node tmp=query(,,n,l,r); printf("%d\n",op==?tmp.sum0:tmp.mx0);}
}
}

2019.04.27