HYSBZ 1858 线段树 区间合并

  1 //Accepted    14560 KB    1532 ms
  2 //线段树 区间合并
  3 /*
  4 0 a b 把[a, b]区间内的所有数全变成0
  5 1 a b 把[a, b]区间内的所有数全变成1
  6 2 a b 把[a,b]区间内的所有数全部取反，也就是说把所有的0变成1，把所有的1变成0
  7 3 a b 询问[a, b]区间内总共有多少个1
  8 4 a b 询问[a, b]区间内最多有多少个连续的1
  9 */
 10 #include <cstdio>
 11 #include <cstring>
 12 #include <iostream>
 13 #include <queue>
 14 #include <cmath>
 15 #include <algorithm>
 16 using namespace std;
 17 /**
 18   * This is a documentation comment block
 19   * 如果有一天你坚持不下去了，就想想你为什么走到这儿！
 20   * @authr songt
 21   */
 22 const int imax_n = 100005;
 23 struct node
 24 {
 25     int l,r;
 26     int L1,R1;   //左右连续1
 27     int L0,R0;   //左右连续0
 28     int change;  //-1 无操作 2取反，0全0 1全1
 29     int sum1,sum0;  //最大连续1,0
 30     int all1,all0;  //所有1,0
 31 }f[imax_n*3];
 32 int a[imax_n];
 33 int max(int a,int b)
 34 {
 35     return a>b?a:b;
 36 }
 37 int min(int a,int b)
 38 {
 39     return a<b?a:b;
 40 }
 41 int swap(int &a,int &b)
 42 {
 43     int t=a;
 44     a=b;
 45     b=t;
 46 }
 47 //由下向上合并
 48 //f[t]的延迟标记是指延迟f[t]的两个孩子的操作,而f[t]已经完成该操作
 49 void pushUp(int t)
 50 {
 51     //printf("pushUpt=%d\n",t);
 52     int lLen=f[2*t].r-f[2*t].l+1;
 53     int rLen=f[2*t+1].r-f[2*t+1].l+1;
 54 
 55     f[t].L1=f[2*t].L1;
 56     if (f[2*t].L1==lLen) f[t].L1+=f[2*t+1].L1;//左连续1的个数
 57     f[t].R1=f[2*t+1].R1;
 58     if (f[2*t+1].R1==rLen) f[t].R1+=f[2*t].R1;//右连续1的个数
 59     f[t].sum1=max(f[2*t].sum1,f[2*t+1].sum1);
 60     f[t].sum1=max(f[t].sum1,f[2*t].R1+f[2*t+1].L1);//最大连续1=max(左边，右边，中间合并)
 61     f[t].all1=f[2*t].all1+f[2*t+1].all1;   //所有1
 62 
 63     f[t].L0=f[2*t].L0;
 64     if (f[2*t].L0==lLen) f[t].L0+=f[2*t+1].L0;
 65     f[t].R0=f[2*t+1].R0;
 66     if (f[2*t+1].R0==rLen) f[t].R0+=f[2*t].R0;
 67     f[t].sum0=max(f[2*t].sum0,f[2*t+1].sum0);
 68     f[t].sum0=max(f[t].sum0,f[2*t].R0+f[2*t+1].L0);
 69     f[t].all0=f[2*t].all0+f[2*t+1].all0;
 70 }
 71 void pushDown(int t)
 72 {
 73     if (f[t].change!=-1)
 74     {
 75         //printf("pushDownt=%d\n",t);
 76         int lLen=f[2*t].r-f[2*t].l+1;
 77         int rLen=f[2*t+1].r-f[2*t+1].l+1;
 78         if (f[t].change==0)  //set all 0
 79         {
 80             f[2*t].change=f[2*t+1].change=0;
 81             f[t].change=-1;
 82             f[2*t].L1=f[2*t].R1=f[2*t].sum1=f[2*t].all1=0;
 83             f[2*t].L0=f[2*t].R0=f[2*t].sum0=f[2*t].all0=lLen;
 84             f[2*t+1].L1=f[2*t+1].R1=f[2*t+1].sum1=f[2*t+1].all1=0;
 85             f[2*t+1].L0=f[2*t+1].R0=f[2*t+1].sum0=f[2*t+1].all0=rLen;
 86             return ;
 87         }
 88         if (f[t].change==1)  //set all 1
 89         {
 90             f[2*t].change=f[2*t+1].change=1;
 91             f[t].change=-1;
 92             f[2*t].L1=f[2*t].R1=f[2*t].sum1=f[2*t].all1=lLen;
 93             f[2*t].L0=f[2*t].R0=f[2*t].sum0=f[2*t].all0=0;
 94             f[2*t+1].L1=f[2*t+1].R1=f[2*t+1].sum1=f[2*t+1].all1=rLen;
 95             f[2*t+1].L0=f[2*t+1].R0=f[2*t+1].sum0=f[2*t+1].all0=0;
 96             return ;
 97         }
 98         if (f[t].change==2)  //0->1 1->0
 99         {
100             f[t].change=-1;
101             if (f[2*t].change==-1)  //如果f[2*t]没有操作，则直接取反
102             {
103                 f[2*t].change=2;
104             }
105             else if (f[2*t].change==0)  //如果f[2*t]已经标记为置0，取反后为置1
106             {
107                 f[2*t].change=1;
108             }
109             else if (f[2*t].change==1)  //如果f[2*t]已经标记为置1，取反后为置0
110             {
111                 f[2*t].change=0;
112             }
113             else if (f[2*t].change==2)  //如果f[2*t]已经取反，再次取反相当于没操作
114             {
115                 f[2*t].change=-1;
116             }
117             swap(f[2*t].L0,f[2*t].L1); //f[2*t]进行取反操作，0,1的标记都要互换
118             swap(f[2*t].R0,f[2*t].R1);
119             swap(f[2*t].sum0,f[2*t].sum1);
120             swap(f[2*t].all0,f[2*t].all1);
121             //2*t+1 同 2*t
122             if (f[2*t+1].change==-1)
123             {
124                 f[2*t+1].change=2;
125             }
126             else if (f[2*t+1].change==0)
127             {
128                 f[2*t+1].change=1;
129             }
130             else if (f[2*t+1].change==1)
131             {
132                 f[2*t+1].change=0;
133             }
134             else if (f[2*t+1].change==2)
135             {
136                 f[2*t+1].change=-1;
137             }
138             swap(f[2*t+1].L0,f[2*t+1].L1);
139             swap(f[2*t+1].R0,f[2*t+1].R1);
140             swap(f[2*t+1].sum0,f[2*t+1].sum1);
141             swap(f[2*t+1].all0,f[2*t+1].all1);
142         }
143     }
144 }
145 void build(int t,int l,int r)
146 {
147     f[t].l=l;
148     f[t].r=r;
149     f[t].change=-1;
150     if (l==r)
151     {
152         f[t].L1=f[t].R1=f[t].sum1=f[t].all1=a[l];
153         f[t].L0=f[t].R0=f[t].sum0=f[t].all0=1-a[l];
154         return ;
155     }
156     int mid=(f[t].l+f[t].r)/2;
157     build(2*t,l,mid);
158     build(2*t+1,mid+1,r);
159     pushUp(t);
160 }
161 void update(int t,int l,int r,int op)
162 {
163     //printf("update l=%d r=%d\n",l,r);
164     if (f[t].l==l && f[t].r==r)
165     {
166         //如果是2号操作，需要考虑原来的操作
167         if (op==2)
168         {
169             if (f[t].change==-1) f[t].change=2;
170             else if (f[t].change==0) f[t].change=1;
171             else if (f[t].change==1) f[t].change=0;
172             else if (f[t].change==2) f[t].change=-1;
173         }
174         else
175         {
176             //置0,1操作和原来的操作没关系
177             f[t].change=op;
178         }
179         if (op==0)  //该区间置0
180         {
181             f[t].L1=f[t].R1=f[t].sum1=f[t].all1=0;
182             f[t].L0=f[t].R0=f[t].sum0=f[t].all0=f[t].r-f[t].l+1;
183         }
184         else if (op==1) //该区间置1
185         {
186             f[t].L1=f[t].R1=f[t].sum1=f[t].all1=f[t].r-f[t].l+1;
187             f[t].L0=f[t].R0=f[t].sum0=f[t].all0=0;
188         }
189         else if (op==2) //该区间取反
190         {
191             swap(f[t].L0,f[t].L1);
192             swap(f[t].R0,f[t].R1);
193             swap(f[t].sum0,f[t].sum1);
194             swap(f[t].all0,f[t].all1);
195         }
196         return ;
197     }
198     pushDown(t); //如果操作的区间为当前区间的子区间，则要把当前区间的change传到子区间
199     int mid=(f[t].l+f[t].r)/2;
200     if (r<=mid) update(2*t,l,r,op);
201     else
202     {
203         if (l>mid) update(2*t+1,l,r,op);
204         else
205         {
206             update(2*t,l,mid,op);
207             update(2*t+1,mid+1,r,op);
208         }
209     }
210     pushUp(t);   //子区间修改完成后，要向父区间合并信息
211 }
212 //op==1 求所有1 op==0 求最大连续1
213 int query(int t,int l,int r,int op)
214 {
215     if (f[t].l==l && f[t].r==r)
216     {
217         if (op==1) return f[t].all1;
218         return f[t].sum1;
219     }
220     pushDown(t);
221     int mid=(f[t].l+f[t].r)/2;
222     if (r<=mid) return query(2*t,l,r,op);
223     else
224     {
225         if (l>mid) return query(2*t+1,l,r,op);
226         else
227         {
228             if (op==1) return query(2*t,l,mid,op)+query(2*t+1,mid+1,r,op);
229             int ans,ans1,ans2;
230             ans1=query(2*t,l,mid,op);
231             ans2=query(2*t+1,mid+1,r,op);
232             ans=max(ans1,ans2);
233             ans=max(ans,min(f[2*t].R1,mid-l+1)+min(f[2*t+1].L1,r-mid));  //r-mid=r-(mid+1)+1
234             return ans;
235         }
236     }
237 }
238 int n,Q;
239 int op,x,y;
240 void slove()
241 {
242     build(1,1,n);
243     for (int i=1;i<=Q;i++)
244     {
245         scanf("%d%d%d",&op,&x,&y);
246         x++;
247         y++;
248         if (op==0)
249         {
250             update(1,x,y,0);
251         }
252         else if (op==1)
253         {
254             update(1,x,y,1);
255         }
256         else if (op==2)
257         {
258             update(1,x,y,2);
259         }
260         else if (op==3)
261         {
262             int t=query(1,x,y,1);
263             printf("%d\n",t);
264         }
265         else if (op==4)
266         {
267             int t=query(1,x,y,0);
268             printf("%d\n",t);
269         }
270     }
271 }
272 int main()
273 {
274     while (scanf("%d%d",&n,&Q)!=EOF)
275     {
276         for (int i=1;i<=n;i++)
277         scanf("%d",&a[i]);
278         slove();
279     }
280     return 0;
281 }