HDU 3397 双lazy标记的问题

题目大意

对一个只有0和1的序列，支持以下几种操作
1.将区间所有的值变成1
2.将区间所有的值变为0
3.将区间的0和1翻转（0变成1 1变成0）
4.求区间中1的个数
5.求区间连续最长的1的个数

http://vjudge.net/problem/viewProblem.action?id=14689

整整一下午。。。简直把自己修改的都要哭了

lazy标记有先后关系，如to[]覆盖后，那么rev[]翻转标记就应该重新赋为0

我们在pushdown中，是对孩子节点进行更新，那么更新的也是孩子节点的lazy标记，to[]覆盖的也是孩子节点的标记，对于还没用过的rev[cur]是不用变的，

这是它父亲传下来的，确保了正确性的

而在update函数中，每次给to[cur]进行了赋值，那么rev[cur]就要重置为0，因为我们这是对当前节点传入的标记，覆盖执行在当前节点上

PS：就是这破玩意改了我一下午的时间

void update(int cur,int x,int y,int s,int t,int op)
{
    int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1;
    if(x>=s&&y<=t){
        if(op==0){
            lc[cur][0]=rc[cur][0]=mc[cur][0]=y-x+1;
            lc[cur][1]=rc[cur][1]=mc[cur][1]=0;
            sum[cur]=0;
            to[cur]=0,rev[cur]=0;
        }
        else if(op==1){
            lc[cur][0]=rc[cur][0]=mc[cur][0]=0;
            lc[cur][1]=rc[cur][1]=mc[cur][1]=y-x+1;
            sum[cur]=y-x+1;
            to[cur]=1,rev[cur]=0;
        }
        else if(op==2){
            swap(lc[cur][0],lc[cur][1]);
            swap(rc[cur][0],rc[cur][1]);
            swap(mc[cur][0],mc[cur][1]);
            sum[cur]=y-x+1-sum[cur];
            rev[cur]^=1;
        }
        return;
    }
    push_down(cur,x,y);
    if(mid>=s) update(L,s,t,op);
    if(mid+1<=t) update(R,s,t,op);
    push_up(cur,x,y);
}

总代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define L ls,x,mid
#define R rs,mid+1,y
#define N 100010
int to[N<<2],rev[N<<2],lc[N<<2][2],rc[N<<2][2],mc[N<<2][2],sum[N<<2],X[N];
void push_up(int cur,int x,int y)
{
    int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1;
    sum[cur]=sum[ls]+sum[rs];
    for(int i=0;i<2;i++){
        lc[cur][i]=lc[ls][i],rc[cur][i]=rc[rs][i];
        mc[cur][i]=max(mc[ls][i],mc[rs][i]);
        mc[cur][i]=max(mc[cur][i],rc[ls][i]+lc[rs][i]);
        if(lc[ls][i]==mid-x+1) lc[cur][i]=lc[ls][i]+lc[rs][i];
        if(rc[rs][i]==y-mid) rc[cur][i]=rc[ls][i]+rc[rs][i];
    }
}
void push_down(int cur,int x,int y)
{
    int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1;
    if(to[cur]!=-1){
        to[ls]=to[rs]=to[cur];
        rev[ls]=rev[rs]=0;
        lc[ls][0]=rc[ls][0]=mc[ls][0]=to[cur]?0:mid-x+1;
        lc[ls][1]=rc[ls][1]=mc[ls][1]=to[cur]?mid-x+1:0;
        lc[rs][0]=rc[rs][0]=mc[rs][0]=to[cur]?0:y-mid;
        lc[rs][1]=rc[rs][1]=mc[rs][1]=to[cur]?y-mid:0;
        sum[ls]=to[cur]*(mid-x+1);
        sum[rs]=to[cur]*(y-mid);
        to[cur]=-1;
    }
    if(rev[cur]){
        rev[ls]^=1,rev[rs]^=1;
        swap(lc[ls][0],lc[ls][1]);
        swap(lc[rs][0],lc[rs][1]);

        swap(rc[ls][0],rc[ls][1]);
        swap(rc[rs][0],rc[rs][1]);

        swap(mc[ls][0],mc[ls][1]);
        swap(mc[rs][0],mc[rs][1]);

        sum[ls]=mid-x+1-sum[ls];
        sum[rs]=y-mid-sum[rs];

        rev[cur]=0;
    }
}
void build(int cur,int x,int y)
{
    int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1;
    to[cur]=-1,rev[cur]=0;
    if(x==y){
        lc[cur][1]=rc[cur][1]=mc[cur][1]=sum[cur]=X[x];
        lc[cur][0]=rc[cur][0]=mc[cur][0]=X[x]^1;
        //printf("%d
",mc[cur][0]);
        return;
    }
    build(L);
    build(R);
    push_up(cur,x,y);
}
void update(int cur,int x,int y,int s,int t,int op)
{
    int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1;
    if(x>=s&&y<=t){
        if(op==0){
            lc[cur][0]=rc[cur][0]=mc[cur][0]=y-x+1;
            lc[cur][1]=rc[cur][1]=mc[cur][1]=0;
            sum[cur]=0;
            to[cur]=0,rev[cur]=0;
        }
        else if(op==1){
            lc[cur][0]=rc[cur][0]=mc[cur][0]=0;
            lc[cur][1]=rc[cur][1]=mc[cur][1]=y-x+1;
            sum[cur]=y-x+1;
            to[cur]=1,rev[cur]=0;
        }
        else if(op==2){
            swap(lc[cur][0],lc[cur][1]);
            swap(rc[cur][0],rc[cur][1]);
            swap(mc[cur][0],mc[cur][1]);
            sum[cur]=y-x+1-sum[cur];
            rev[cur]^=1;
        }
        return;
    }
    push_down(cur,x,y);
    if(mid>=s) update(L,s,t,op);
    if(mid+1<=t) update(R,s,t,op);
    push_up(cur,x,y);
}
void query1(int cur,int x,int y,int s,int t,int &ans)
{
    int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1;
    if(x>=s&&y<=t){
        ans+=sum[cur];
        return;
    }
    push_down(cur,x,y);
    if(mid>=s) query1(L,s,t,ans);
    if(mid+1<=t) query1(R,s,t,ans);
}
int query2(int cur,int x,int y,int s,int t)
{
    int mid=(x+y)/2,ls=cur<<1,rs=cur<<1|1;
    if(x>=s&&y<=t){
        return mc[cur][1];
    }
    push_down(cur,x,y);
    int ans=0;
    if(mid>=s) ans=max(ans,query2(L,s,t));
    if(mid+1<=t) ans=max(ans,query2(R,s,t));
    return max(ans,min(mid-s+1,rc[ls][1])+min(t-mid,lc[rs][1]));
}
int main()
{
    int T,n,m,op,a,b;
    scanf("%d",&T);
    while(T--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++) scanf("%d",&X[i]);
        build(1,1,n);
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&op,&a,&b);
            if(op==3){
                int ans=0;
                query1(1,1,n,a+1,b+1,ans);
                printf("%d
",ans);
            }
            else if(op==4){
                printf("%d
",query2(1,1,n,a+1,b+1));
            }
            else update(1,1,n,a+1,b+1,op);
        }
    }
    return 0;
}