HDU 3397 线段树区间修改

Posted 2020-10-01 zzq

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  三维表示区间对应的结点值。

  1 #include<bits/stdc++.h>
  2 using namespace std;
  3 #define MAXN ((100000<<2)+15)
  4 int g[3][3]={0,0,0,1,1,1,1,0,-1};
  5 struct SegmentTree
  6 {
  7     #define M ((L+R)>>1)
  8     #define lc (id<<1)
  9     #define rc (id<<1|1)
 10     int num[MAXN][2],X;
 11     int len[2][3][MAXN];
 12     short int laz[MAXN];
 13     void show(int L,int R,int id)
 14     {
 15         if(L==R) {cout<<num[id][1]<<" ";return;}
 16         push_down(L,R,id);
 17         show(L,M,lc);
 18         show(M+1,R,rc);
 19         push_up(L,R,id);
 20     }
 21     void init()
 22     {
 23         memset(num,0,sizeof(num));
 24         memset(laz,-1,sizeof(laz));
 25         memset(len,0,sizeof(len));
 26     }
 27     void push_up(int L,int R,int id)
 28     {
 29        num[id][0]=num[lc][0]+num[rc][0];
 30        num[id][1]=num[lc][1]+num[rc][1];
 31        for(int i=0;i<=1;++i){
 32         len[i][0][id]=len[i][0][lc];
 33         len[i][1][id]=len[i][1][rc];
 34         len[i][2][id]=max(max(len[i][2][lc],len[i][2][rc]),len[i][1][lc]+len[i][0][rc]);
 35         if(!num[lc][i^1]) len[i][0][id]=max(len[i][0][id],len[i][0][lc]+len[i][0][rc]);
 36         if(!num[rc][i^1]) len[i][1][id]=max(len[i][1][id],len[i][1][rc]+len[i][1][lc]);
 37        }
 38     }
 39     void push_down(int L,int R,int id)
 40     {
 41         if(laz[id]==-1) return;
 42           laz[lc]=laz[lc]==-1?laz[id]:g[laz[id]][laz[lc]];
 43           laz[rc]=laz[rc]==-1?laz[id]:g[laz[id]][laz[rc]];
 44         if(laz[id]<2){
 45             num[lc][laz[id]^1]=num[rc][laz[id]^1]=0;
 46             num[lc][laz[id]]=M-L+1;
 47             num[rc][laz[id]]=R-M;
 48             len[laz[id]^1][0][lc]=len[laz[id]^1][1][lc]=len[laz[id]^1][2][lc]=0;
 49             len[laz[id]^1][0][rc]=len[laz[id]^1][1][rc]=len[laz[id]^1][2][rc]=0;
 50             len[laz[id]][0][lc]=len[laz[id]][1][lc]=len[laz[id]][2][lc]=M-L+1;
 51             len[laz[id]][0][rc]=len[laz[id]][1][rc]=len[laz[id]][2][rc]=R-M;
 52         }
 53         else {
 54                 swap(num[lc][0],num[lc][1]);
 55                 swap(num[rc][0],num[rc][1]);
 56                 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]);
 57                 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]);
 58         }
 59           laz[id]=-1;
 60     }
 61     void build(int L,int R,int id)
 62     {
 63         if(L==R)
 64         {
 65             scanf("%d",&X);num[id][X]++;
 66             len[X][2][id]=len[X][0][id]=len[X][1][id]=1;
 67             return;
 68         }
 69         build(L,M,lc);
 70         build(M+1,R,rc);
 71         push_up(L,R,id);
 72     }
 73     void change(int L,int R,int id,int l,int r,int x)
 74     {
 75         if(L>=l&&R<=r){
 76             num[id][x^1]=0;
 77             num[id][x]=R-L+1;
 78             len[x][0][id]=len[x][1][id]=len[x][2][id]=R-L+1;
 79             len[x^1][0][id]=len[x^1][1][id]=len[x^1][2][id]=0;
 80             laz[id]=laz[id]==-1?x:g[x][laz[id]];
 81             return;
 82         }
 83         push_down(L,R,id);
 84         if(l<=M) change(L,M,lc,l,r,x);
 85         if(r>M) change(M+1,R,rc,l,r,x);
 86         push_up(L,R,id);
 87     }
 88     void swp(int L,int R,int id,int l,int r)
 89     {
 90         if(L>=l&&R<=r){
 91             swap(num[id][0],num[id][1]);
 92             laz[id]=laz[id]==-1?2:g[2][laz[id]];
 93             swap(len[0][0][id],len[1][0][id]);
 94             swap(len[0][1][id],len[1][1][id]);
 95             swap(len[0][2][id],len[1][2][id]);
 96             return;
 97         }
 98         push_down(L,R,id);
 99         if(l<=M) swp(L,M,lc,l,r);
100         if(r>M)  swp(M+1,R,rc,l,r);
101         push_up(L,R,id);
102     }
103     int num1(int L,int R,int id,int l,int r)
104     {
105         if(L>=l&&R<=r) return num[id][1];
106         push_down(L,R,id);
107         int s=0;
108         if(l<=M) s+=num1(L,M,lc,l,r);
109         if(r>M)  s+=num1(M+1,R,rc,l,r);
110         push_up(L,R,id);
111         return s;
112     }
113     int mx1(int L,int R,int id,int l,int r)
114     {
115         if(L>=l&&R<=r) return len[1][2][id];
116         push_down(L,R,id);
117         int s=0;
118         if(r<=M) s=max(s, mx1(L,M,lc,l,r));
119         else if(l>M) s=max(s,mx1(M+1,R,rc,l,r));
120         else {
121                 int lm=len[1][1][lc],rm=len[1][0][rc];
122         if(lm>M-l+1) lm=M-l+1;
123         if(rm>r-M) rm=r-M;
124                 s=max(s,max(max(mx1(L,M,lc,l,r),mx1(M+1,R,rc,l,r)),lm+rm));
125         }
126         push_up(L,R,id);
127         return s;
128     }
129 }seg;
130 
131 int main()
132 {
133     int t,n,m,i,j,k,opt,a,b;
134     scanf("%d",&t);
135     while(t--){
136         scanf("%d%d",&n,&m);
137         seg.init();
138         seg.build(1,n,1);
139     while(m--){
140         scanf("%d%d%d",&opt,&a,&b);a++;b++;
141         if(opt==0)        seg.change(1,n,1,a,b,0);
142         else if(opt==1)   seg.change(1,n,1,a,b,1);
143         else if(opt==2)   seg.swp(1,n,1,a,b);
144         else if(opt==3)   printf("%d\n",seg.num1(1,n,1,a,b));
145         else              printf("%d\n",seg.mx1(1,n,1,a,b));
146     }
147     }
148     return 0;
149 }