吉首大学2019年程序设计竞赛（重现赛）-K(线段树)

题目链接：https://ac.nowcoder.com/acm/contest/992/K

题意：给一个大小为1e5的数组，由0 1组成，有两种操作，包括区间修改，将一段区间内的0换成1，1换成0; 区间查询，查询区间内连续1的数量。

思路：区间查询和区间修改，明显可以用线段树来做，我们先分析下复杂度，每次操作复杂度为logn，有m次操作(m<=1e5+1)，那么总复杂度为mlogn，是可行的。

　　　我们用线段树维护什么呢，因为要求区间最多连续1的个数。那么需要维护区间前缀连续0/1的个数fr[0]/fr[1]，区间后缀0/1的个数ba[0]/ba[1]，区间连续0/1个数最大值mx[0]/mx[1]。维护连续0的个数是因为0和1可以相互转换，这样进行更新时就方便不少。懒惰标记lazy可以用异或1来更新，因为更新两次就相当与不更新。

　　　因为维护的元素较多，pushup和pushdown操作就稍微复杂。这题查询的方式是需要学习的，返回区间最大连续1的个数，注意考虑由左区间后缀1和右区间前缀1组合的情况。

详见代码：

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;

const int maxn=;

struct node{
    int l,r,len,fr[],ba[],mx[];
    int lazy;
}tr[maxn<<];

int n,m,ans,a[maxn];

void pushup(int v){
    if(tr[v<<].fr[]==tr[v<<].len){
        tr[v].fr[]=tr[v<<].fr[]+tr[v<<|].fr[];
        tr[v].fr[]=;
    }
    else if(tr[v<<].fr[]==tr[v<<].len){
        tr[v].fr[]=tr[v<<].fr[]+tr[v<<|].fr[];
        tr[v].fr[]=;
    }
    else{
        tr[v].fr[]=tr[v<<].fr[];
        tr[v].fr[]=tr[v<<].fr[];
    }
    if(tr[v<<|].ba[]==tr[v<<|].len){
        tr[v].ba[]=tr[v<<].ba[]+tr[v<<|].ba[];
        tr[v].ba[]=;
    }
    else if(tr[v<<|].ba[]==tr[v<<|].len){
        tr[v].ba[]=tr[v<<].ba[]+tr[v<<|].ba[];
        tr[v].ba[]=;
    }
    else{
        tr[v].ba[]=tr[v<<|].ba[];
        tr[v].ba[]=tr[v<<|].ba[];
    }
    tr[v].mx[]=max(tr[v<<].ba[]+tr[v<<|].fr[],max(tr[v<<].mx[],tr[v<<|].mx[]));
    tr[v].mx[]=max(tr[v<<].ba[]+tr[v<<|].fr[],max(tr[v<<].mx[],tr[v<<|].mx[]));
}

void pushdown(int v){
    tr[v<<].lazy^=,tr[v<<|].lazy^=;
    swap(tr[v<<].fr[],tr[v<<].fr[]);
    swap(tr[v<<].ba[],tr[v<<].ba[]);
    swap(tr[v<<].mx[],tr[v<<].mx[]);
    swap(tr[v<<|].fr[],tr[v<<|].fr[]);
    swap(tr[v<<|].ba[],tr[v<<|].ba[]);
    swap(tr[v<<|].mx[],tr[v<<|].mx[]);
    tr[v].lazy=;
}

void build(int v,int l,int r){
    tr[v].l=l,tr[v].r=r,tr[v].len=r-l+;
    if(l==r){
        if(a[r]){
            tr[v].fr[]=,tr[v].fr[]=;
            tr[v].ba[]=,tr[v].ba[]=;
            tr[v].mx[]=,tr[v].mx[]=;
        }
        else{
            tr[v].fr[]=,tr[v].fr[]=;
            tr[v].ba[]=,tr[v].ba[]=;
            tr[v].mx[]=,tr[v].mx[]=;
        }
        return;
    }
    int mid=(l+r)>>;
    build(v<<,l,mid);
    build(v<<|,mid+,r);
    pushup(v);
}

void update(int v,int l,int r){
    if(l<=tr[v].l&&r>=tr[v].r){
        swap(tr[v].fr[],tr[v].fr[]);
        swap(tr[v].ba[],tr[v].ba[]);
        swap(tr[v].mx[],tr[v].mx[]);
        tr[v].lazy^=;
        return;
    }
    if(tr[v].lazy) pushdown(v);
    int mid=(tr[v].l+tr[v].r)>>;
    if(l<=mid) update(v<<,l,r);
    if(r>mid) update(v<<|,l,r);
    pushup(v);
}

int query(int v,int l,int r){
    if(tr[v].l==l&&tr[v].r==r)
        return tr[v].mx[];
    if(tr[v].lazy) pushdown(v);
    int mid=(tr[v].l+tr[v].r)>>;
    if(r<=mid)
        return query(v<<,l,r);
    else if(l>mid)
        return query(v<<|,l,r);
    else{
        int x,y,z;
        x=min(tr[v<<].ba[],mid-l+)+min(tr[v<<|].fr[],r-mid);
        y=query(v<<,l,mid);
        z=query(v<<|,mid+,r);
        return max(x,max(y,z));
    }
}

int main(){
    scanf("%d",&n);
    for(int i=;i<=n;++i)
        scanf("%d",&a[i]);
    build(,,n);
    scanf("%d",&m);
    while(m--){
        int op,x,y;
        scanf("%d%d%d",&op,&x,&y);
        if(op==)
            update(,x,y);
        else
            printf("%d\n",query(,x,y));
    }
    return ;
}