[BZOJ1858][Scoi2010]序列操作 线段树

链接

题意：5种操作

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

题解

线段树上维护 \(cnt[0/1]\) 表示 \(0/1\) 的个数， \(lmax[0/1],rmax[0/1],tmax[0/1]\) 表示前缀最大个数，后缀最大个数，总的最大个数。

#include<bits/stdc++.h>
#define REP(i,a,b) for(int i(a);i<=(b);++i)
using namespace std;
typedef long long ll;
inline int read(){char c;int w;
    while(!isdigit(c=getchar()));w=c&15;
    while(isdigit(c=getchar()))w=w*10+(c&15);return w;
}
inline char smax(int&x,const int&y){return x<y?x=y,1:0;}
inline char smin(int&x,const int&y){return x>y?x=y,1:0;}
const int N=1e5+5;
struct data{int len,cnt,lmax,rmax,tmax;};
inline data merge(const data&a,const data&b){
    data c;c.cnt=a.cnt+b.cnt;c.len=a.len+b.len;
    c.lmax=a.cnt==a.len?a.lmax+b.lmax:a.lmax;
    c.rmax=b.cnt==b.len?a.rmax+b.rmax:b.rmax;
    c.tmax=max(a.tmax,b.tmax);
    smax(c.tmax,a.rmax+b.lmax);
    return c;
}
struct node{data c[2];bool rev;int len,set;}t[N<<2];
#define ls o<<1
#define rs o<<1|1
inline void change(int o,int w){
    if(!o)return;t[o].rev=0,t[o].set=w;int&len=t[o].len;
    t[o].c[w]=(data){len,len,len,len,len};
    t[o].c[!w]=(data){len,0,0,0,0};
}
inline void rever(int o){
    if(!o)return;
    if(~t[o].set)change(o,!t[o].set);
    else t[o].rev^=1,swap(t[o].c[0],t[o].c[1]);
    
}
#define pushup() \
    t[o].c[0]=merge(t[ls].c[0],t[rs].c[0]),\
    t[o].c[1]=merge(t[ls].c[1],t[rs].c[1])
#define pushdown() \
    if(~t[o].set)change(ls,t[o].set),change(rs,t[o].set),t[o].set=-1;\
    if(t[o].rev)rever(ls),rever(rs),t[o].rev=0;

inline data ask(int o,int l,int r,int x,int y){
    if(x<=l&&r<=y)return t[o].c[1];
    int mid=l+r>>1;pushdown();
    if(y<=mid)return ask(ls,l,mid,x,y);
    if(x>mid)return ask(rs,mid+1,r,x,y);
    return merge(ask(ls,l,mid,x,y),ask(rs,mid+1,r,x,y));
}
inline void update(int o,int l,int r,int x,int y){
    if(x<=l&&r<=y){rever(o);return;}
    int mid=l+r>>1;pushdown();
    if(x<=mid)update(ls,l,mid,x,y);
    if(y>mid)update(rs,mid+1,r,x,y);
    pushup();
}
inline void update(int o,int l,int r,int x,int y,int z){
    if(x<=l&&r<=y){change(o,z);return;}
    int mid=l+r>>1;pushdown();
    if(x<=mid)update(ls,l,mid,x,y,z);
    if(y>mid)update(rs,mid+1,r,x,y,z);
    pushup();
}
int n,m,a[N];
inline void build(int o,int l,int r){
    t[o].set=-1;t[o].rev=0;t[o].len=r-l+1;
    if(l==r){change(o,a[l]);return;}
    int mid=l+r>>1;build(ls,l,mid),build(rs,mid+1,r);
    pushup();
}
int main(){
    n=read();m=read();
    #define all 1,1,n
    REP(i,1,n)a[i]=read();build(all);
    while(m--){
        int op=read(),a=read()+1,b=read()+1;
        switch(op){
            case 0:update(all,a,b,0);break;
            case 1:update(all,a,b,1);break;
            case 2:update(all,a,b);break;
            case 3:printf("%d\n",ask(all,a,b).cnt);break;
            case 4:printf("%d\n",ask(all,a,b).tmax);break;
        }
    //  REP(i,1,n)printf("%d ",ask(all,i,i).cnt);puts("");
    }
    return 0;
}

转载于:https://www.cnblogs.com/HolyK/p/9884645.html