线段树 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<<2];  
  
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+1-T[u].sum;  
}  
  
void PushUp(int u){  
    if(T[u].lef==T[u].rig)return;  
    int len=T[u].rig-T[u].lef+1;  
    T[u].lsum=T[lson].lsum;  
    T[u].rsum=T[rson].rsum;  
    if(T[u].lsum==(len+1)>>1)T[u].lsum+=T[rson].lsum;  
    if(T[u].rsum==len>>1)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+1)>>1)T[u].lz+=T[rson].lz;  
    if(T[u].rz==len>>1)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!=-1){  
        int len=T[u].rig-T[u].lef+1;  
        T[lson].COVER=T[rson].COVER=T[u].COVER;  
        T[lson].XOR=T[rson].XOR=0;  
        T[lson].lsum=T[lson].rsum=T[lson].msum=T[u].COVER?(len+1)>>1:0;  
        T[lson].lz=T[lson].rz=T[lson].mz=T[u].COVER?0:(len+1)>>1;  
        T[lson].sum=T[u].COVER?(len+1)>>1:0;  
  
        T[rson].lsum=T[rson].rsum=T[rson].msum=T[u].COVER?len>>1:0;  
        T[rson].lz=T[rson].rz=T[rson].mz=T[u].COVER?0:len>>1;  
        T[rson].sum=T[u].COVER?len>>1:0;  
  
        T[u].COVER=-1;  
    }  
    if(T[u].XOR){  
        T[u].XOR=0;  
        T[lson].XOR^=1;  
        T[rson].XOR^=1;  
        makeXOR(lson);  
        makeXOR(rson);  
    }  
}  
  
void Build(int u,int l,int r){  
    T[u].lef=l;  
    T[u].rig=r;  
    T[u].COVER=-1;  
    T[u].XOR=0;  
    if(l==r){  
        T[u].lsum=T[u].rsum=T[u].msum=(dat[l]==1);  
        T[u].lz=T[u].rz=T[u].mz=(dat[l]==0);  
        T[u].sum=dat[l];  
        T[u].COVER=dat[l];  
        return;  
    }  
    int mid=(l+r)>>1;  
    Build(lson,l,mid);  
    Build(rson,mid+1,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<2){//覆盖操作  
            int len=T[u].rig-T[u].lef+1;  
            T[u].COVER=op;  
            T[u].lsum=T[u].rsum=T[u].msum=op?len:0;  
            T[u].lz=T[u].rz=T[u].mz=op?0:len;  
            T[u].sum=op?len:0;  
        }  
        else {  
            T[u].XOR=1;  
            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==3)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==3) 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+1)+min(T[rson].lsum,r-T[rson].lef+1);  
        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=1;i<=n;i++)scanf("%d",dat+i);  
        Build(1,1,n);  
  
        while(m--){  
            scanf("%d%d%d",&cmd,&a,&b);  
            a++,b++;  
            if(cmd<3)Update(1,a,b,cmd);  
            else printf("%d\n",Query(1,a,b,cmd));  
        }  
    }  
    return 0;  
}