HDU 3397 线段树区间修改

Sequence operation

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 8308    Accepted Submission(s): 2507

Problem Description

lxhgww got a sequence contains n characters which are all '0's or '1's.
We have five operations here:
Change operations:
0 a b change all characters into '0's in [a , b]
1 a b change all characters into '1's in [a , b]
2 a b change all '0's into '1's and change all '1's into '0's in [a, b]
Output operations:
3 a b output the number of '1's in [a, b]
4 a b output the length of the longest continuous '1' string in [a , b]

 

Input

T(T<=10) in the first line is the case number.
Each case has two integers in the first line: n and m (1 <= n , m <= 100000).
The next line contains n characters, '0' or '1' separated by spaces.
Then m lines are the operations:
op a b: 0 <= op <= 4 , 0 <= a <= b < n.

 

Output

For each output operation , output the result.

 

Sample Input

1 10 10 0 0 0 1 1 0 1 0 1 1 1 0 2 3 0 5 2 2 2 4 0 4 0 3 6 2 3 7 4 2 8 1 0 5 0 5 6 3 3 9

 

Sample Output

5 2 6 5

 

Author

lxhgww&&shǎ崽

及其恶心的一道题目，要维护许多变量= =

由于mx1函数写错找bug找了半天，最后发现自己的思路错了诶，可算A了。

中间维护的变量: 1的个数，0的个数，以第一个开始的最长连续0，以第一个开始的最长连续1，以最后一个开始的最长连续0，以最后一个开始的最长连续1，

区间最长连续1，区间最长连续0，可能有一些变量是互补的我写的有点多余不过感觉很方便把看起来，另外注意当操作为0/1时可以覆盖之前所有的标记。

我用0，1,2表示三种标记全0全1和翻转。len储存区间信息 一维表示'0'/'1'   二维表示 head，tail，max_continue_1  三维表示区间对应的结点值。

#include<bits/stdc++.h>
using namespace std;
#define MAXN ((100000<<2)+15)
int g[3][3]={0,0,0,1,1,1,1,0,-1};
struct SegmentTree
{
    #define M ((L+R)>>1)
    #define lc (id<<1)
    #define rc (id<<1|1)
    int num[MAXN][2],X;
    int len[2][3][MAXN];
    short int laz[MAXN];
    void show(int L,int R,int id)
    {
        if(L==R) {cout<<num[id][1]<<" ";return;}
        push_down(L,R,id);
        show(L,M,lc);
        show(M+1,R,rc);
        push_up(L,R,id);
    }
    void init()
    {
        memset(num,0,sizeof(num));
        memset(laz,-1,sizeof(laz));
        memset(len,0,sizeof(len));
    }
    void push_up(int L,int R,int id)
    {
       num[id][0]=num[lc][0]+num[rc][0];
       num[id][1]=num[lc][1]+num[rc][1];
       for(int i=0;i<=1;++i){
        len[i][0][id]=len[i][0][lc];
        len[i][1][id]=len[i][1][rc];
        len[i][2][id]=max(max(len[i][2][lc],len[i][2][rc]),len[i][1][lc]+len[i][0][rc]);
        if(!num[lc][i^1]) len[i][0][id]=max(len[i][0][id],len[i][0][lc]+len[i][0][rc]);
        if(!num[rc][i^1]) len[i][1][id]=max(len[i][1][id],len[i][1][rc]+len[i][1][lc]);
       }
    }
    void push_down(int L,int R,int id)
    {
        if(laz[id]==-1) return;
          laz[lc]=laz[lc]==-1?laz[id]:g[laz[id]][laz[lc]];
          laz[rc]=laz[rc]==-1?laz[id]:g[laz[id]][laz[rc]];
        if(laz[id]<2){
            num[lc][laz[id]^1]=num[rc][laz[id]^1]=0;
            num[lc][laz[id]]=M-L+1;
            num[rc][laz[id]]=R-M;
            len[laz[id]^1][0][lc]=len[laz[id]^1][1][lc]=len[laz[id]^1][2][lc]=0;
            len[laz[id]^1][0][rc]=len[laz[id]^1][1][rc]=len[laz[id]^1][2][rc]=0;
            len[laz[id]][0][lc]=len[laz[id]][1][lc]=len[laz[id]][2][lc]=M-L+1;
            len[laz[id]][0][rc]=len[laz[id]][1][rc]=len[laz[id]][2][rc]=R-M;
        }
        else {
                swap(num[lc][0],num[lc][1]);
                swap(num[rc][0],num[rc][1]);
                swap(len[0][0][lc],len[1][0][lc]); swap(len[0][1][lc],len[1][1][lc]); swap(len[0][2][lc],len[1][2][lc]);
                swap(len[0][0][rc],len[1][0][rc]); swap(len[0][1][rc],len[1][1][rc]); swap(len[0][2][rc],len[1][2][rc]);
        }
          laz[id]=-1;
    }
    void build(int L,int R,int id)
    {
        if(L==R)
        {
            scanf("%d",&X);num[id][X]++;
            len[X][2][id]=len[X][0][id]=len[X][1][id]=1;
            return;
        }
        build(L,M,lc);
        build(M+1,R,rc);
        push_up(L,R,id);
    }
    void change(int L,int R,int id,int l,int r,int x)
    {
        if(L>=l&&R<=r){
            num[id][x^1]=0;
            num[id][x]=R-L+1;
            len[x][0][id]=len[x][1][id]=len[x][2][id]=R-L+1;
            len[x^1][0][id]=len[x^1][1][id]=len[x^1][2][id]=0;
            laz[id]=laz[id]==-1?x:g[x][laz[id]];
            return;
        }
        push_down(L,R,id);
        if(l<=M) change(L,M,lc,l,r,x);
        if(r>M) change(M+1,R,rc,l,r,x);
        push_up(L,R,id);
    }
    void swp(int L,int R,int id,int l,int r)
    {
        if(L>=l&&R<=r){
            swap(num[id][0],num[id][1]);
            laz[id]=laz[id]==-1?2:g[2][laz[id]];
            swap(len[0][0][id],len[1][0][id]);
            swap(len[0][1][id],len[1][1][id]);
            swap(len[0][2][id],len[1][2][id]);
            return;
        }
        push_down(L,R,id);
        if(l<=M) swp(L,M,lc,l,r);
        if(r>M)  swp(M+1,R,rc,l,r);
        push_up(L,R,id);
    }
    int num1(int L,int R,int id,int l,int r)
    {
        if(L>=l&&R<=r) return num[id][1];
        push_down(L,R,id);
        int s=0;
        if(l<=M) s+=num1(L,M,lc,l,r);
        if(r>M)  s+=num1(M+1,R,rc,l,r);
        push_up(L,R,id);
        return s;
    }
    int mx1(int L,int R,int id,int l,int r)
    {
        if(L>=l&&R<=r) return len[1][2][id];
        push_down(L,R,id);
        int s=0;
        if(r<=M) s=max(s, mx1(L,M,lc,l,r));
        else if(l>M) s=max(s,mx1(M+1,R,rc,l,r));
        else {
                int lm=len[1][1][lc],rm=len[1][0][rc];
        if(lm>M-l+1) lm=M-l+1;
        if(rm>r-M) rm=r-M;
                s=max(s,max(max(mx1(L,M,lc,l,r),mx1(M+1,R,rc,l,r)),lm+rm));
        }
        push_up(L,R,id);
        return s;
    }
}seg;

int main()
{
    int t,n,m,i,j,k,opt,a,b;
    scanf("%d",&t);
    while(t--){
        scanf("%d%d",&n,&m);
        seg.init();
        seg.build(1,n,1);
    while(m--){
        scanf("%d%d%d",&opt,&a,&b);a++;b++;
        if(opt==0)        seg.change(1,n,1,a,b,0);
        else if(opt==1)   seg.change(1,n,1,a,b,1);
        else if(opt==2)   seg.swp(1,n,1,a,b);
        else if(opt==3)   printf("%d
",seg.num1(1,n,1,a,b));
        else              printf("%d
",seg.mx1(1,n,1,a,b));
    }
    }
    return 0;
}