线段树 HDU 3397

5种操作

具体看代码

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;  

#define lson u<<1
#define rson u<<1|1
#define MAXN 100010  

int dat[MAXN];  

struct Node {
    int lef,rig;
    int lsum,rsum,msum;//左端最长连续1，右端最长连续1，区间最长连续1
    int lz,rz,mz;//左端最长连续0，右端最长连续0，区间最长连续0
    int sum;//区间1的个数
    int COVER,XOR;
}T[MAXN<<];  

void makeXOR(int u){//翻转操作
    swap(T[u].lsum,T[u].lz);
    swap(T[u].rsum,T[u].rz);
    swap(T[u].msum,T[u].mz);
    T[u].sum=T[u].rig-T[u].lef+-T[u].sum;
}  

void PushUp(int u){
    if(T[u].lef==T[u].rig)return;
    int len=T[u].rig-T[u].lef+;
    T[u].lsum=T[lson].lsum;
    T[u].rsum=T[rson].rsum;
    if(T[u].lsum==(len+)>>)T[u].lsum+=T[rson].lsum;
    if(T[u].rsum==len>>)T[u].rsum+=T[lson].rsum;
    T[u].msum=max(T[lson].msum,T[rson].msum);
    T[u].msum=max(T[u].msum,T[lson].rsum+T[rson].lsum);  

    T[u].lz=T[lson].lz;
    T[u].rz=T[rson].rz;
    if(T[u].lz==(len+)>>)T[u].lz+=T[rson].lz;
    if(T[u].rz==len>>)T[u].rz+=T[lson].rz;
    T[u].mz=max(T[lson].mz,T[rson].mz);
    T[u].mz=max(T[u].mz,T[lson].rz+T[rson].lz);  

    T[u].sum=T[lson].sum+T[rson].sum;
}  

void PushDown(int u){
    if(T[u].lef==T[u].rig)return;
    if(T[u].COVER!=-){
        int len=T[u].rig-T[u].lef+;
        T[lson].COVER=T[rson].COVER=T[u].COVER;
        T[lson].XOR=T[rson].XOR=;
        T[lson].lsum=T[lson].rsum=T[lson].msum=T[u].COVER?(len+)>>:;
        T[lson].lz=T[lson].rz=T[lson].mz=T[u].COVER?:(len+)>>;
        T[lson].sum=T[u].COVER?(len+)>>:;  

        T[rson].lsum=T[rson].rsum=T[rson].msum=T[u].COVER?len>>:;
        T[rson].lz=T[rson].rz=T[rson].mz=T[u].COVER?:len>>;
        T[rson].sum=T[u].COVER?len>>:;  

        T[u].COVER=-;
    }
    if(T[u].XOR){
        T[u].XOR=;
        T[lson].XOR^=;
        T[rson].XOR^=;
        makeXOR(lson);
        makeXOR(rson);
    }
}  

void Build(int u,int l,int r){
    T[u].lef=l;
    T[u].rig=r;
    T[u].COVER=-;
    T[u].XOR=;
    if(l==r){
        T[u].lsum=T[u].rsum=T[u].msum=(dat[l]==);
        T[u].lz=T[u].rz=T[u].mz=(dat[l]==);
        T[u].sum=dat[l];
        T[u].COVER=dat[l];
        return;
    }
    int mid=(l+r)>>;
    Build(lson,l,mid);
    Build(rson,mid+,r);
    PushUp(u);
}  

void Update(int u,int l,int r,int op){
    PushDown(u);//这里应该先pushdown
    if(l<=T[u].lef&&T[u].rig<=r){
        if(op<){//覆盖操作
            int len=T[u].rig-T[u].lef+;
            T[u].COVER=op;
            T[u].lsum=T[u].rsum=T[u].msum=op?len:;
            T[u].lz=T[u].rz=T[u].mz=op?:len;
            T[u].sum=op?len:;
        }
        else {
            T[u].XOR=;
            makeXOR(u);
        }
    }
    else {
        if(l<=T[lson].rig)Update(lson,l,r,op);
        if(r>=T[rson].lef)Update(rson,l,r,op);
        PushUp(u);
    }
}  

int Query(int u,int l,int r,int op){
    PushDown(u);//应该先 pushdown，以防错误
    if(l<=T[u].lef&&T[u].rig<=r){
        if(op==)return T[u].sum;
        else return T[u].msum;
    }
    else {
        if(r<=T[lson].rig)return Query(lson,l,r,op);
        if(l>=T[rson].lef)return Query(rson,l,r,op);
        if(op==) return Query(lson,l,T[lson].rig,op)+Query(rson,T[rson].lef,r,op);  

        int ret=min(T[lson].rsum,T[lson].rig-l+)+min(T[rson].lsum,r-T[rson].lef+);
        int ans=max(Query(lson,l,T[lson].rig,op),Query(rson,T[rson].lef,r,op));
        return max(ans,ret);  

    }
}  

int main(){  

    int t;
    scanf("%d",&t);
    while(t--){
        int n,m;
        int cmd,a,b;  

        scanf("%d%d",&n,&m);
        for(int i=;i<=n;i++)scanf("%d",dat+i);
        Build(,,n);  

        while(m--){
            scanf("%d%d%d",&cmd,&a,&b);
            a++,b++;
            if(cmd<)Update(,a,b,cmd);
            else printf("%d\n",Query(,a,b,cmd));
        }
    }
    return ;
}