bzoj1858 [Scoi2010]序列操作

题目链接

恶心线段树

既然要翻转我就对0、1分别用一个information结构体存信息，翻转就直接swap

注意标记的优先次序。。。(坑

  1 #include<algorithm>
  2 #include<iostream>
  3 #include<cstdlib>
  4 #include<cstring>
  5 #include<cstdio>
  6 #include<string>
  7 #include<cmath>
  8 #include<ctime>
  9 #include<queue>
 10 #include<stack>
 11 #include<map>
 12 #include<set>
 13 #define rre(i,r,l) for(int i=(r);i>=(l);i--)
 14 #define re(i,l,r) for(int i=(l);i<=(r);i++)
 15 #define Clear(a,b) memset(a,b,sizeof(a))
 16 #define inout(x) printf("%d",(x))
 17 #define douin(x) scanf("%lf",&x)
 18 #define strin(x) scanf("%s",(x))
 19 #define LLin(x) scanf("%lld",&x)
 20 #define op operator
 21 #define CSC main
 22 typedef unsigned long long ULL;
 23 typedef const int cint;
 24 typedef long long LL;
 25 using namespace std;
 26 void inin(int &ret)
 27 {
 28     ret=0;int f=0;char ch=getchar();
 29     while(ch<'0'||ch>'9'){if(ch=='-')f=1;ch=getchar();}
 30     while(ch>='0'&&ch<='9')ret*=10,ret+=ch-'0',ch=getchar();
 31     ret=f?-ret:ret;
 32 }
 33 int n,m;
 34 struct information
 35 {
 36     int Max,maxl,maxr,sum;
 37     information(){Max=maxl=maxr=sum=0;}
 38     information(int Max,int maxl,int maxr,int sum):Max(Max),maxl(maxl),maxr(maxr),sum(sum){}
 39 };
 40 struct segmenttree
 41 {
 42     int l,r,change;
 43     information in[2];
 44     segmenttree(){change=-1;}
 45 }t[400010];
 46 int a[100010];
 47 void update(int k)
 48 {
 49     int p1=k<<1,p2=p1|1,l=t[k].l,r=t[k].r,mid=(l+r)>>1;
 50     re(i,0,1)
 51     {
 52         t[k].in[i].Max=
 53             max(t[p1].in[i].maxr+t[p2].in[i].maxl,
 54             max(t[p1].in[i].Max, t[p2].in[i].Max));
 55         t[k].in[i].maxl=
 56             max(t[p1].in[i].maxl,
 57             (t[p1].in[i].maxl==mid-l+1?mid-l+1+t[p2].in[i].maxl:0));
 58         t[k].in[i].maxr=
 59             max(t[p2].in[i].maxr,
 60             (t[p2].in[i].maxr==r-mid?r-mid+t[p1].in[i].maxr:0));
 61         t[k].in[i].sum=t[p1].in[i].sum+t[p2].in[i].sum;
 62     }
 63 }
 64 void build(int k,int l,int r)
 65 {
 66     t[k].l=l,t[k].r=r;
 67     if(l==r){t[k].in[a[l]]=information(1,1,1,1);return ;}
 68     int mid=(l+r)>>1;
 69     build(k<<1,l,mid);
 70     build(k<<1|1,mid+1,r);
 71     update(k);
 72 }
 73 void down(int k)
 74 {
 75     int p1=k<<1,p2=k<<1|1;
 76     int x=t[k].change,y=x^1,len1=t[p1].r-t[p1].l+1,len2=t[p2].r-t[p2].l+1;
 77     if(x!=-1)
 78         if(x!=2)
 79         {
 80             t[p1].change=t[p2].change=x;
 81             t[p1].in[x]=information(len1,len1,len1,len1);
 82             t[p1].in[y]=information(0,0,0,0);
 83             t[p2].in[x]=information(len2,len2,len2,len2);
 84             t[p2].in[y]=information(0,0,0,0);
 85             t[k].change=-1;
 86         }
 87         else
 88         {
 89             t[k].change=-1;
 90             swap(t[p1].in[0],t[p1].in[1]);
 92                 if(t[p1].change!=-1)
 93                     if(t[p1].change==2)t[p1].change=-1;else t[p1].change^=1;
 95                 else t[p1].change=2;
 97             swap(t[p2].in[0],t[p2].in[1]);
 99                 if(t[p2].change!=-1)
100                     if(t[p2].change==2)t[p2].change=-1;else t[p2].change^=1;
102                 else t[p2].change=2;
103         }else ;
105     update(k);
106 }
107 void change(int k,int l,int r,int x)
108 {
109     if(t[k].l>=l&&t[k].r<=r)
110     {
111         t[k].change=x;
112         int y=x^1,len=t[k].r-t[k].l+1;
113         t[k].in[x]=information(len,len,len,len);
114         t[k].in[y]=information(0,0,0,0);
115         return ;
116     }
117     down(k);
118     int mid=(t[k].l+t[k].r)>>1,p1=k<<1,p2=p1|1;
119     if(r<=mid)change(p1,l,r,x);
120     else if(l>mid)change(p2,l,r,x);
121     else change(p1,l,r,x),change(p2,l,r,x);
122     update(k);
123 }
124 void change2(int k,int l,int r)
125 {
126     if(t[k].l>=l&&t[k].r<=r)
127     {
128         swap(t[k].in[0],t[k].in[1]);
129         if(t[k].change!=-1)
130             if(t[k].change==2)t[k].change=-1;
131             else t[k].change^=1;
132         else t[k].change=2;
133         return ;
134     }
135     down(k);
136     int mid=(t[k].l+t[k].r)>>1,p1=k<<1,p2=p1|1;
137     if(r<=mid)change2(p1,l,r);
138     else if(l>mid)change2(p2,l,r);
139     else change2(p1,l,r),change2(p2,l,r);
140     update(k);
141 }
142 int queryl(int k,int l,int r)
143 {
144     if(t[k].l>=l&&t[k].r<=r)return t[k].in[1].maxl;
145     down(k);
146     int mid=(t[k].l+t[k].r)>>1,p1=k<<1,p2=p1|1;
147     if(r<=mid)return queryl(p1,l,r);
148     if(l>mid)return queryl(p2,l,r);
149     int ret=queryl(p1,l,r);if(ret==t[p1].r-t[p1].l+1)ret+=queryl(p2,l,r);
150     return ret;
151 }
152 int queryr(int k,int l,int r)
153 {
154     if(t[k].l>=l&&t[k].r<=r)return t[k].in[1].maxr;
155     down(k);
156     int mid=(t[k].l+t[k].r)>>1,p1=k<<1,p2=p1|1;
157     if(r<=mid)return queryr(p1,l,r);
158     if(l>mid)return queryr(p2,l,r);
159     int ret=queryr(p2,l,r);if(ret==t[p2].r-t[p2].l+1)ret+=queryr(p1,l,r);
160     return ret;
161 }
162 int query2(int k,int l,int r)
163 {
164     if(t[k].l>=l&&t[k].r<=r)return t[k].in[1].Max;
165     down(k);
166     int mid=(t[k].l+t[k].r)>>1,p1=k<<1,p2=p1|1;
167     if(r<=mid)return query2(p1,l,r);
168     if(l>mid)return query2(p2,l,r);
169     return max(max(query2(p1,l,r),query2(p2,l,r)),queryl(p2,l,r)+queryr(p1,l,r));
170 }
171 int query(int k,int l,int r)
172 {
173     if(t[k].l>=l&&t[k].r<=r)return t[k].in[1].sum;
174     down(k);
175     int mid=(t[k].l+t[k].r)>>1,p1=k<<1,p2=p1|1;
176     if(r<=mid)return query(p1,l,r);
177     if(l>mid)return query(p2,l,r);
178     return query(p1,l,r)+query(p2,l,r);
179 }
188 int CSC()
189 {
190     freopen("in.in","r",stdin);
191     freopen("out.out","w",stdout);
192     inin(n),inin(m);
193     re(i,1,n)inin(a[i]);
194     build(1,1,n);
195     re(i,1,m)
196     {
197         int opt,l,r;
198         inin(opt),inin(l),inin(r);l++,r++;
199         if(opt==0)change(1,l,r,0);
200         if(opt==1)change(1,l,r,1);
201         if(opt==2)change2(1,l,r);
202         if(opt==3)cout<<query(1,l,r)<<"\n";
203         if(opt==4)cout<<query2(1,l,r)<<"\n";
206     }
207     return 0;
208 }